Refactor env to add key word arguments from config yaml (#223)

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-05-30 13:45:22 +02:00
parent 2c2e4e14ed
commit 265b0ec44d
7 changed files with 59 additions and 36 deletions
--- a/lerobot/configs/default.yaml
+++ b/lerobot/configs/default.yaml
@@ -37,6 +37,8 @@ training:
  save_freq: ???
  log_freq: 250
  save_checkpoint: true
+  num_workers: 4
+  batch_size: ???

 eval:
  n_episodes: 1