opentau.utils
Utility functions and helpers for the OpenTau package.
This module provides various utility functions for logging, random number generation, training utilities, device management, and other common operations used throughout the OpenTau codebase.
Modules
Utilities for managing Accelerate accelerator instances. |
|
Utilities for benchmarking and timing code execution. |
|
Utilities for working with PyTorch FakeTensor. |
|
Utilities for interacting with the Hugging Face Hub. |
|
Utilities for checking package availability and versions. |
|
Utilities for file I/O operations. |
|
Utilities for recording and converting LIBERO datasets. |
|
Utilities for tracking and logging training metrics. |
|
Monkey patches to inject behaviour into computational graph construction. |
|
Utilities for managing random number generator states. |
|
ROS 2 to LeRobot dataset extractors and utilities. |
|
Utilities for training checkpoint management and state persistence. |
|
Module for patching transformers |
|
General utility functions for device management, logging, and common operations. |
|
Vision-tower helpers reused across policies (pi06, pi07, ...). |