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

revert changes to default.yaml

This commit is contained in:
Alexander Soare 2024-03-21 10:27:07 +00:00
parent 72d3c3120b
commit 4e10cd306b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/configs/default.yaml
View File

@ -12,14 +12,14 @@ hydra:
seed: 1337 seed: 1337
# batch size for TorchRL SerialEnv. Each underlying env will get the seed = seed + env_index # batch size for TorchRL SerialEnv. Each underlying env will get the seed = seed + env_index
# NOTE: only diffusion policy supports rollout_batch_size > 1 # NOTE: only diffusion policy supports rollout_batch_size > 1
rollout_batch_size: 10 rollout_batch_size: 1
device: cuda # cpu device: cuda # cpu
prefetch: 4 prefetch: 4
eval_freq: ??? eval_freq: ???
save_freq: ??? save_freq: ???
eval_episodes: ??? eval_episodes: ???
save_video: false save_video: false
save_model: true save_model: false
save_buffer: false save_buffer: false
train_steps: ??? train_steps: ???
fps: ??? fps: ???
@ -34,6 +34,6 @@ policy: ???
wandb: wandb:
enable: true enable: true
# Set to true to disable saving an artifact despite save_model == True # Set to true to disable saving an artifact despite save_model == True
disable_artifact: true disable_artifact: false
project: lerobot project: lerobot
notes: "" notes: ""