First real commit, simxarm env added with torchrl!

This commit is contained in:
Cadene
2024-01-29 12:49:30 +00:00
parent 0396980450
commit 1144819c29
11 changed files with 437 additions and 1 deletions

16
lerobot/scripts/train.py Normal file
View File

@@ -0,0 +1,16 @@
import hydra
import torch
from termcolor import colored
from ..lib.utils import set_seed
@hydra.main(version_base=None, config_name="default", config_path="../configs")
def train(cfg: dict):
assert torch.cuda.is_available()
set_seed(cfg.seed)
print(colored("Work dir:", "yellow", attrs=["bold"]), cfg.log_dir)
if __name__ == "__main__":
train()