v0.7 (#26)

* update from scratch configs

* update gym pretraining configs - use fewer epochs

* update robomimic pretraining configs - use fewer epochs

* allow trajectory plotting in eval agent

* add simple vit unet

* update avoid pretraining configs - use fewer epochs

* update furniture pretraining configs - use same amount of epochs as before

* add robomimic diffusion unet pretraining configs

* update robomimic finetuning configs - higher lr

* add vit unet checkpoint urls

* update pretraining and finetuning instructions as configs are updated

cfg/gym/pretrain/hopper-medium-v2/pre_diffusion_mlp.yaml

@@ -24,12 +24,12 @@ wandb:
   run: ${now:%H-%M-%S}_${name}
 
 train:
-  n_epochs: 3000
+  n_epochs: 200
   batch_size: 128
   learning_rate: 1e-3
   weight_decay: 1e-6
   lr_scheduler:
-    first_cycle_steps: 3000
+    first_cycle_steps: 200
     warmup_steps: 1
     min_lr: 1e-4
   save_model_freq: 100

cfg/gym/pretrain/hopper-medium-v2/pre_gaussian_mlp.yaml

@@ -23,12 +23,12 @@ wandb:
   run: ${now:%H-%M-%S}_${name}
 
 train:
-  n_epochs: 500
+  n_epochs: 200
   batch_size: 128
   learning_rate: 1e-4
   weight_decay: 1e-6
   lr_scheduler:
-    first_cycle_steps: 1000
+    first_cycle_steps: 200
     warmup_steps: 1
     min_lr: 1e-4
   save_model_freq: 100