Upload folder using huggingface_hub

config.json

@@ -6,8 +6,8 @@
   "clip_sample_range": 1.0,
   "crop_is_random": true,
   "crop_shape": [
-    84,
-    84
+    448,
+    448
   ],
   "diffusion_step_embed_dim": 128,
   "do_mask_loss_for_padding": false,

config.yaml

@@ -5,13 +5,13 @@ seed: 100000
 dataset_repo_id: tinkerbuggy/sugar_can_coffee_cup
 video_backend: pyav
 training:
-  offline_steps: 2000
+  offline_steps: 4000
   num_workers: 4
   batch_size: 64
   eval_freq: -1
   log_freq: 200
   save_checkpoint: true
-  save_freq: 300
+  save_freq: 25000
   online_steps: 0
   online_rollout_n_episodes: 1
   online_rollout_batch_size: 1
@@ -51,14 +51,14 @@ training:
       - 0.8
       - 1.2
   grad_clip_norm: 10
-  lr: 0.001
+  lr: 0.0001
   lr_scheduler: cosine
   lr_warmup_steps: 500
   adam_betas:
   - 0.95
   - 0.999
-  adam_eps: 1.0e-06
-  adam_weight_decay: 0.0001
+  adam_eps: 1.0e-08
+  adam_weight_decay: 1.0e-06
   delta_timestamps:
     observation.images.laptop:
     - -0.03333333333333333
@@ -132,8 +132,8 @@ policy:
     action: min_max
   vision_backbone: resnet18
   crop_shape:
-  - 84
-  - 84
+  - 448
+  - 448
   crop_is_random: true
   pretrained_backbone_weights: null
   use_group_norm: true

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eef7188079ffeac28475df52349ef6d5a6f1111cb282b8e348df17850fd48486
-size 1051837392
+oid sha256:a25cc8a173be15299b6d245e02e5282fc2afe1d267e0bb90977f3ab28cf5f17c
+size 1051838888