Upload policy weights, train config and readme

config.json

@@ -34,7 +34,7 @@
         }
     },
     "device": "cuda",
-    "use_amp": true,
+    "use_amp": false,
     "use_peft": false,
     "push_to_hub": true,
     "repo_id": "annyi/record-green-cube",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f8b7c23b89320c246959c4e78a4b8ffa6681e22c45a3b402f2bf52021dc66a8f
+oid sha256:b2c17f410f42704bc3d5731a03535dc7b9161accb5f904c857f25b8d404bde61
 size 206699736

train_config.json

@@ -116,7 +116,7 @@
             }
         },
         "device": "cuda",
-        "use_amp": true,
+        "use_amp": false,
         "use_peft": false,
         "push_to_hub": true,
         "repo_id": "annyi/record-green-cube",
@@ -151,7 +151,7 @@
         "optimizer_weight_decay": 0.0001,
         "optimizer_lr_backbone": 1e-05
     },
-    "output_dir": "outputs/train/so101_act_pick_green_cube_amp",
+    "output_dir": "outputs/train/so101_act_pick_green_cube",
     "job_name": "so101_act_pick_green_cube",
     "resume": false,
     "seed": 1000,