forked from tangger/lerobot
First real commit, simxarm env added with torchrl!
This commit is contained in:
16
lerobot/scripts/train.py
Normal file
16
lerobot/scripts/train.py
Normal 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()
|
||||
Reference in New Issue
Block a user