Upload policy weights, train config and readme

config.json

@@ -26,7 +26,7 @@
         }
     },
     "device": "cuda",
-    "use_amp": false,
+    "use_amp": true,
     "use_peft": false,
     "push_to_hub": true,
     "repo_id": "open-cloth/diffusion-cloth",
@@ -42,7 +42,7 @@
         "ACTION": "MIN_MAX"
     },
     "drop_n_last_frames": 7,
-    "vision_backbone": "resnet50",
+    "vision_backbone": "resnet18",
     "resize_shape": [
         240,
         320
@@ -53,14 +53,14 @@
         288
     ],
     "crop_is_random": true,
-    "pretrained_backbone_weights": "ResNet50_Weights.IMAGENET1K_V1",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "use_group_norm": false,
     "spatial_softmax_num_keypoints": 64,
     "use_separate_rgb_encoder_per_camera": false,
     "down_dims": [
+        256,
         512,
-        1024,
-        2048
+        1024
     ],
     "kernel_size": 5,
     "n_groups": 8,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e047487260f13fe3ed35bedf45640555f439282fc90fb192c8db78007e72a1c2
-size 1116593992
+oid sha256:5721932efb6c34a538d253578375c75fd0f2d79390fd8564dfa061940df0a96e
+size 312295392

train_config.json

@@ -108,7 +108,7 @@
             }
         },
         "device": "cuda",
-        "use_amp": false,
+        "use_amp": true,
         "use_peft": false,
         "push_to_hub": true,
         "repo_id": "open-cloth/diffusion-cloth",
@@ -124,7 +124,7 @@
             "ACTION": "MIN_MAX"
         },
         "drop_n_last_frames": 7,
-        "vision_backbone": "resnet50",
+        "vision_backbone": "resnet18",
         "resize_shape": [
             240,
             320
@@ -135,14 +135,14 @@
             288
         ],
         "crop_is_random": true,
-        "pretrained_backbone_weights": "ResNet50_Weights.IMAGENET1K_V1",
+        "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
         "use_group_norm": false,
         "spatial_softmax_num_keypoints": 64,
         "use_separate_rgb_encoder_per_camera": false,
         "down_dims": [
+            256,
             512,
-            1024,
-            2048
+            1024
         ],
         "kernel_size": 5,
         "n_groups": 8,
@@ -175,7 +175,7 @@
     "resume": false,
     "seed": 1000,
     "cudnn_deterministic": false,
-    "num_workers": 4,
+    "num_workers": 22,
     "batch_size": 64,
     "steps": 200,
     "eval_freq": 20000,
@@ -211,7 +211,7 @@
         "project": "robot-learning",
         "entity": null,
         "notes": null,
-        "run_id": "um0p4cfp",
+        "run_id": "njbxluis",
         "mode": null,
         "add_tags": true
     },