Upload checkpoint from step 065670

config.json

@@ -33,7 +33,7 @@
     "private": null,
     "tags": null,
     "license": null,
-    "pretrained_path": null,
+    "pretrained_path": "outputs/train/diffusion_no_aug/checkpoints/last/pretrained_model",
     "horizon": 64,
     "n_action_steps": 32,
     "normalization_mapping": {

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebc04ca28ceb1e0429319e44c30d2a8de704d4d7615de35cd01820753fdc228e
+oid sha256:d30813c7a39af3018037020774deced715fe41222fc205a8b637012d61087c8f
 size 327664088

train_config.json

@@ -115,7 +115,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": null,
+        "pretrained_path": "outputs/train/diffusion_no_aug/checkpoints/last/pretrained_model",
         "horizon": 64,
         "n_action_steps": 32,
         "normalization_mapping": {
@@ -172,12 +172,12 @@
     },
     "output_dir": "outputs/train/diffusion_no_aug",
     "job_name": "diffusion-v2-no-aug",
-    "resume": false,
+    "resume": true,
     "seed": 1000,
     "cudnn_deterministic": false,
     "num_workers": 8,
     "batch_size": 64,
-    "steps": 98505,
+    "steps": 65670,
     "eval_freq": 20000,
     "log_freq": 3,
     "tolerance_s": 0.0001,
@@ -221,8 +221,6 @@
     "rabc_kappa": 0.01,
     "rabc_epsilon": 1e-06,
     "rabc_head_mode": "sparse",
-    "rename_map": {
-        "observation.images.gripper": "observation.images.camera1"
-    },
-    "checkpoint_path": null
+    "rename_map": {},
+    "checkpoint_path": "outputs/train/diffusion_no_aug/checkpoints/last"
 }