tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Simplify optim imports

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 12:27:20 +01:00
parent ec2990518b
commit 00698305a3
3 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/optim/__init__.py
View File

@@ -12,4 +12,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from .optimizers import OptimizerConfig as OptimizerConfig from .optimizers import OptimizerConfig
from .schedulers import LRSchedulerConfig
__all__ = ["OptimizerConfig", "LRSchedulerConfig"]

3
lerobot/common/policies/config.py
View File

@@ -23,8 +23,7 @@ from huggingface_hub import hf_hub_download
from huggingface_hub.constants import CONFIG_NAME from huggingface_hub.constants import CONFIG_NAME
from huggingface_hub.errors import HfHubHTTPError from huggingface_hub.errors import HfHubHTTPError
from lerobot.common.optim.optimizers import OptimizerConfig from lerobot.common.optim import LRSchedulerConfig, OptimizerConfig
from lerobot.common.optim.schedulers import LRSchedulerConfig
from lerobot.common.utils.hub import HubMixin from lerobot.common.utils.hub import HubMixin
from lerobot.common.utils.utils import auto_select_torch_device, is_amp_available, is_torch_device_available from lerobot.common.utils.utils import auto_select_torch_device, is_amp_available, is_torch_device_available
from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature from lerobot.configs.types import FeatureType, NormalizationMode, PolicyFeature

3
lerobot/configs/train.py
View File

@@ -22,8 +22,7 @@ from huggingface_hub import hf_hub_download
from huggingface_hub.errors import HfHubHTTPError from huggingface_hub.errors import HfHubHTTPError
from lerobot.common import envs from lerobot.common import envs
from lerobot.common.optim import OptimizerConfig from lerobot.common.optim import LRSchedulerConfig, OptimizerConfig
from lerobot.common.optim.schedulers import LRSchedulerConfig
from lerobot.common.policies import PreTrainedConfig from lerobot.common.policies import PreTrainedConfig
from lerobot.common.utils.hub import HubMixin from lerobot.common.utils.hub import HubMixin
from lerobot.configs import parser from lerobot.configs import parser