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

backup wip

Browse Source
This commit is contained in:
Alexander Soare
2024-04-05 18:46:30 +01:00
parent ecc7dd3b17
commit 8d2463f45b
5 changed files with 105 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/eval.py
View File

@@ -251,7 +251,7 @@ def eval(cfg: dict, out_dir=None, stats_path=None):
dataset = make_dataset(cfg, stats_path=stats_path)
logging.info("Making environment.")
env = make_env(cfg, num_parallel_envs=cfg.eval_episodes)
env = make_env(cfg, num_parallel_envs=cfg.rollout_batch_size)
# when policy is None, rollout a random policy
policy = make_policy(cfg) if cfg.policy.pretrained_model_path else None

3
lerobot/scripts/train.py
View File

@@ -148,7 +148,8 @@ def train(cfg: dict, out_dir=None, job_name=None):
# )
logging.info("make_env")
env = make_env(cfg, num_parallel_envs=cfg.eval_episodes)
# TODO(now): uncomment
#env = make_env(cfg, num_parallel_envs=cfg.eval_episodes)
logging.info("make_policy")
policy = make_policy(cfg)