hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

ready for review

Browse Source
This commit is contained in:
Alexander Soare
2024-04-12 11:36:52 +01:00
parent 5666ec3ec7
commit 6d0a45a97d
7 changed files with 11 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/3_train_policy.py
View File

@@ -32,8 +32,6 @@ policy = DiffusionPolicy(
cfg=cfg.policy,
cfg_device=cfg.device,
cfg_noise_scheduler=cfg.noise_scheduler,
cfg_rgb_model=cfg.rgb_model,
cfg_obs_encoder=cfg.obs_encoder,
cfg_optimizer=cfg.optimizer,
cfg_ema=cfg.ema,
**cfg.policy,