From dc745e3037289a8c93cde66523e801a688321eac Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 27 Mar 2024 13:05:13 +0000 Subject: [PATCH] Remove unused part of diffusion policy config --- lerobot/configs/policy/diffusion.yaml | 26 -------------------------- 1 file changed, 26 deletions(-) diff --git a/lerobot/configs/policy/diffusion.yaml b/lerobot/configs/policy/diffusion.yaml index acb368ed..4d6eedca 100644 --- a/lerobot/configs/policy/diffusion.yaml +++ b/lerobot/configs/policy/diffusion.yaml @@ -103,29 +103,3 @@ optimizer: betas: [0.95, 0.999] eps: 1.0e-8 weight_decay: 1.0e-6 - -training: - device: "cuda:0" - seed: 42 - debug: False - resume: True - # optimization - # lr_scheduler: cosine - # lr_warmup_steps: 500 - num_epochs: 8000 - # gradient_accumulate_every: 1 - # EMA destroys performance when used with BatchNorm - # replace BatchNorm with GroupNorm. - # use_ema: True - freeze_encoder: False - # training loop control - # in epochs - rollout_every: 50 - checkpoint_every: 50 - val_every: 1 - sample_every: 5 - # steps per epoch - max_train_steps: null - max_val_steps: null - # misc - tqdm_interval_sec: 1.0