wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

backup wip

This commit is contained in:
Alexander Soare 2024-05-09 13:53:20 +01:00
parent cce0c53d95
commit ad49a4f3f5
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/configs/policy/tdmpc.yaml
View File

@ -58,7 +58,7 @@ policy:
discount: 0.9 discount: 0.9
# Inference. # Inference.
use_mpc: false use_mpc: true
cem_iterations: 6 cem_iterations: 6
max_std: 2.0 max_std: 2.0
min_std: 0.05 min_std: 0.05

1
lerobot/scripts/train.py
View File

@ -303,6 +303,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
init_logging() init_logging()
if cfg.training.online_steps > 0 and cfg.eval.batch_size > 1: if cfg.training.online_steps > 0 and cfg.eval.batch_size > 1:
# TODO(now)
logging.warning("eval.batch_size > 1 not supported for online training steps") logging.warning("eval.batch_size > 1 not supported for online training steps")
# Check device is available # Check device is available