Refactor eval.py (#127)

2024-05-03 17:33:16 +01:00
parent b7b69fcc3d
commit bccee745c3
12 changed files with 457 additions and 298 deletions
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -28,6 +28,7 @@ training:

 eval:
  n_episodes: 50
+  batch_size: 50

 override_dataset_stats:
  # TODO(rcadene, alexander-soare): should we remove image stats as well? do we use a pretrained vision model?