Upload policy weights, train config and readme

README.md

@@ -5,9 +5,9 @@ license: apache-2.0
 model_name: groot
 pipeline_tag: robotics
 tags:
+- groot
 - lerobot
 - robotics
-- groot
 ---
 
 # Model Card for groot

config.json

@@ -61,7 +61,7 @@
     "tune_llm": false,
     "tune_visual": false,
     "tune_projector": true,
-    "tune_diffusion_model": false,
+    "tune_diffusion_model": true,
     "lora_rank": 0,
     "lora_alpha": 16,
     "lora_dropout": 0.1,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:01ad6c78cb4f8ccda2c110d7604757d1f8cc1fd19a0bf4c1389189860ecf7e2f
+oid sha256:4cdc44d78ca6530882b377751fd83f2f6d30984880ae0ed93f63740ca7eba3fa
 size 6964802952

train_config.json

@@ -143,7 +143,7 @@
         "tune_llm": false,
         "tune_visual": false,
         "tune_projector": true,
-        "tune_diffusion_model": false,
+        "tune_diffusion_model": true,
         "lora_rank": 0,
         "lora_alpha": 16,
         "lora_dropout": 0.1,
@@ -175,7 +175,7 @@
     "seed": 1000,
     "num_workers": 4,
     "batch_size": 32,
-    "steps": 4000,
+    "steps": 10000,
     "eval_freq": 20000,
     "log_freq": 200,
     "tolerance_s": 0.0001,
@@ -211,7 +211,7 @@
         "project": "dynamic_object_v0",
         "entity": "jeongeun",
         "notes": null,
-        "run_id": "kepepi04",
+        "run_id": "wz59zrao",
         "mode": null
     },
     "peft": null,