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

Merge remote-tracking branch 'Cadene/user/rcadene/2024_03_31_remove_torchrl' into refactor_act_remove_torchrl

Browse Source
This commit is contained in:
Alexander Soare
2024-04-05 12:00:31 +01:00
parent 9d77f5773d ad3379a73a
commit 4863e54ce9
3 changed files with 28 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lerobot/scripts/train.py
View File

@@ -1,5 +1,4 @@
import logging
from itertools import cycle
from pathlib import Path
import hydra
@@ -7,10 +6,16 @@ import numpy as np
import torch
from lerobot.common.datasets.factory import make_dataset
from lerobot.common.datasets.utils import cycle
from lerobot.common.envs.factory import make_env
from lerobot.common.logger import Logger, log_output_dir
from lerobot.common.policies.factory import make_policy
from lerobot.common.utils import format_big_number, get_safe_torch_device, init_logging, set_global_seed
from lerobot.common.utils import (
format_big_number,
get_safe_torch_device,
init_logging,
set_global_seed,
)
from lerobot.scripts.eval import eval_policy