Upload checkpoint from step 032836

config.json

@@ -33,7 +33,7 @@
     "private": null,
     "tags": null,
     "license": null,
-    "pretrained_path": null,
+    "pretrained_path": "outputs/train/diffusion/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:a59df10639a3ba727aacc21ecd4dfd9cca9c2d66566e26e00f57085a6aec46e3
+oid sha256:7d68417b8d18579898057e6f575d5b3d769915999107b64dbbba44e52169dc83
 size 327664088

train_config.json

@@ -125,7 +125,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": null,
+        "pretrained_path": "outputs/train/diffusion/checkpoints/last/pretrained_model",
         "horizon": 64,
         "n_action_steps": 32,
         "normalization_mapping": {
@@ -182,12 +182,12 @@
     },
     "output_dir": "outputs/train/diffusion",
     "job_name": "diffusion-v2",
-    "resume": false,
+    "resume": true,
     "seed": 1000,
     "cudnn_deterministic": false,
-    "num_workers": 8,
+    "num_workers": 12,
     "batch_size": 64,
-    "steps": 98505,
+    "steps": 65670,
     "eval_freq": 20000,
     "log_freq": 3,
     "tolerance_s": 0.0001,
@@ -231,8 +231,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/checkpoints/last"
 }