From 4e10cd306b00e07b06b0b761f1d42579182f6db7 Mon Sep 17 00:00:00 2001 From: Alexander Soare <alexander.soare159@gmail.com> Date: Thu, 21 Mar 2024 10:27:07 +0000 Subject: [PATCH] revert changes to default.yaml --- lerobot/configs/default.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lerobot/configs/default.yaml b/lerobot/configs/default.yaml index 90d4c06b..52fd1d60 100644 --- a/lerobot/configs/default.yaml +++ b/lerobot/configs/default.yaml @@ -12,14 +12,14 @@ hydra: seed: 1337 # batch size for TorchRL SerialEnv. Each underlying env will get the seed = seed + env_index # NOTE: only diffusion policy supports rollout_batch_size > 1 -rollout_batch_size: 10 +rollout_batch_size: 1 device: cuda # cpu prefetch: 4 eval_freq: ??? save_freq: ??? eval_episodes: ??? save_video: false -save_model: true +save_model: false save_buffer: false train_steps: ??? fps: ??? @@ -34,6 +34,6 @@ policy: ??? wandb: enable: true # Set to true to disable saving an artifact despite save_model == True - disable_artifact: true + disable_artifact: false project: lerobot notes: ""