danbhf commited on Jan 29

Commit

7938897

verified ·

1 Parent(s): 80b3ad4

Upload folder using huggingface_hub

Browse files

This view is limited to 50 files because it contains too many changes. See raw diff

Files changed (50) hide show

checkpoint_005000/config.json +77 -0
checkpoint_005000/model.safetensors +3 -0
checkpoint_005000/policy_postprocessor.json +32 -0
checkpoint_005000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_005000/policy_preprocessor.json +70 -0
checkpoint_005000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_005000/training_metadata.json +36 -0
checkpoint_005000/training_state.pt +3 -0
checkpoint_010000/config.json +77 -0
checkpoint_010000/model.safetensors +3 -0
checkpoint_010000/policy_postprocessor.json +32 -0
checkpoint_010000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_010000/policy_preprocessor.json +70 -0
checkpoint_010000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_010000/training_metadata.json +36 -0
checkpoint_010000/training_state.pt +3 -0
checkpoint_015000/config.json +77 -0
checkpoint_015000/model.safetensors +3 -0
checkpoint_015000/policy_postprocessor.json +32 -0
checkpoint_015000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_015000/policy_preprocessor.json +70 -0
checkpoint_015000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_015000/training_metadata.json +36 -0
checkpoint_015000/training_state.pt +3 -0
checkpoint_020000/config.json +77 -0
checkpoint_020000/model.safetensors +3 -0
checkpoint_020000/policy_postprocessor.json +32 -0
checkpoint_020000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_020000/policy_preprocessor.json +70 -0
checkpoint_020000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_020000/training_metadata.json +36 -0
checkpoint_020000/training_state.pt +3 -0
checkpoint_025000/config.json +77 -0
checkpoint_025000/model.safetensors +3 -0
checkpoint_025000/policy_postprocessor.json +32 -0
checkpoint_025000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_025000/policy_preprocessor.json +70 -0
checkpoint_025000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_025000/training_metadata.json +36 -0
checkpoint_025000/training_state.pt +3 -0
checkpoint_030000/config.json +77 -0
checkpoint_030000/model.safetensors +3 -0
checkpoint_030000/policy_postprocessor.json +32 -0
checkpoint_030000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
checkpoint_030000/policy_preprocessor.json +70 -0
checkpoint_030000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
checkpoint_030000/training_metadata.json +36 -0
checkpoint_030000/training_state.pt +3 -0
checkpoint_035000/config.json +77 -0
checkpoint_035000/model.safetensors +3 -0

checkpoint_005000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_005000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:70bf999ba1c1eec7050b8e6eac0af03f4c0034216aa057cf42a43064d7f27637
+size 206699736

checkpoint_005000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_005000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_005000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_005000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_005000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_005000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:af7a46bae81550e5568d3f2ca9caab98c74199482d2dc012a62b7ff0f7694ca8
+size 412908960

checkpoint_010000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_010000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:f2483a0ee39c1c5d41a2d3bf529c66a9db089fa625384741353c17735c05d0bd
+size 206699736

checkpoint_010000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_010000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_010000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_010000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_010000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_010000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:149f80fcf0394682a2999d714bae676faa47d4ff45fc0efd7cee38be7783c934
+size 412908960

checkpoint_015000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_015000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:721e9618d89012b43ce39992304f149d5e019052f265cfed2ead1baa3d2155ac
+size 206699736

checkpoint_015000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_015000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_015000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_015000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_015000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_015000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:ef47992d02e97bad61a73c0f8e309bb5992854113637b33259587a22b4f5ca1d
+size 412908960

checkpoint_020000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_020000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:d6dd1d0caf7c56b41f04239dd3ee055c6f9a7ac0bcc01e8a22b6dbddac310ca7
+size 206699736

checkpoint_020000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_020000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_020000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_020000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_020000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_020000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:906bdc6e819b00798957efa05bb0322ab74adaf27d3d1fd122cd5c2d9338ed23
+size 412908960

checkpoint_025000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_025000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:1207fd65a4d88442158a6a7636f4eac06d564ce50f5e9acec036950807439a13
+size 206699736

checkpoint_025000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_025000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_025000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_025000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_025000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_025000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:4512077f830a32c7b6afbc78ff58c0d4f9490ccfc30e6f5ab16252931b8447b6
+size 412908960

checkpoint_030000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_030000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:75f47391b271b566480229618af56e2cb5fc1d0dc40edbf2b7ca8205e0f1ad9d
+size 206699736

checkpoint_030000/policy_postprocessor.json ADDED Viewed

	@@ -0,0 +1,32 @@

+{
+  "name": "policy_postprocessor",
+  "steps": [
+    {
+      "registry_name": "unnormalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cpu",
+        "float_dtype": null
+      }
+    }
+  ]
+}

checkpoint_030000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_030000/policy_preprocessor.json ADDED Viewed

	@@ -0,0 +1,70 @@

+{
+  "name": "policy_preprocessor",
+  "steps": [
+    {
+      "registry_name": "rename_observations_processor",
+      "config": {
+        "rename_map": {}
+      }
+    },
+    {
+      "registry_name": "to_batch_processor",
+      "config": {}
+    },
+    {
+      "registry_name": "device_processor",
+      "config": {
+        "device": "cuda",
+        "float_dtype": null
+      }
+    },
+    {
+      "registry_name": "normalizer_processor",
+      "config": {
+        "eps": 1e-08,
+        "features": {
+          "observation.state": {
+            "type": "STATE",
+            "shape": [
+              6
+            ]
+          },
+          "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+              3,
+              480,
+              640
+            ]
+          },
+          "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+              2
+            ]
+          },
+          "action": {
+            "type": "ACTION",
+            "shape": [
+              6
+            ]
+          }
+        },
+        "norm_map": {
+          "VISUAL": "MEAN_STD",
+          "STATE": "MEAN_STD",
+          "ACTION": "MEAN_STD"
+        }
+      },
+      "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
+    }
+  ]
+}

checkpoint_030000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:731ccfe10cfcaec1785fa0d13ef572a513213a2b13bd23abd9bac9015a1bc887
+size 1384

checkpoint_030000/training_metadata.json ADDED Viewed

	@@ -0,0 +1,36 @@

+{
+  "dataset_repo_id": "danbhf/sim_pick_place_157ep",
+  "scene": "so101_with_wrist_cam.xml",
+  "scene_cameras": {
+    "overhead_cam": {
+      "fovy": 52.0,
+      "pos": [
+        0.3,
+        -0.09,
+        0.6
+      ]
+    },
+    "wrist_cam": {
+      "fovy": 71.0,
+      "pos": [
+        0.0,
+        0.0,
+        0.0
+      ]
+    }
+  },
+  "cameras": [
+    "wrist_cam",
+    "overhead_cam"
+  ],
+  "camera_resolutions": {
+    "wrist_cam": "640x480",
+    "overhead_cam": "640x480"
+  },
+  "action_space": "joint (6-dim: normalized joints)",
+  "action_dim": 6,
+  "chunk_size": 100,
+  "fps": 30,
+  "total_frames": 22534,
+  "pickup_coords": true
+}

checkpoint_030000/training_state.pt ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:e4396e6ea8e58de6ac3e60ba2f750d79805065a05876d94f7d13cde1135754a4
+size 412908960

checkpoint_035000/config.json ADDED Viewed

	@@ -0,0 +1,77 @@

+{
+    "type": "act",
+    "n_obs_steps": 1,
+    "input_features": {
+        "observation.state": {
+            "type": "STATE",
+            "shape": [
+                6
+            ]
+        },
+        "observation.images.wrist_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.images.overhead_cam": {
+            "type": "VISUAL",
+            "shape": [
+                3,
+                480,
+                640
+            ]
+        },
+        "observation.environment_state": {
+            "type": "STATE",
+            "shape": [
+                2
+            ]
+        }
+    },
+    "output_features": {
+        "action": {
+            "type": "ACTION",
+            "shape": [
+                6
+            ]
+        }
+    },
+    "device": "cuda",
+    "use_amp": false,
+    "use_peft": false,
+    "push_to_hub": true,
+    "repo_id": null,
+    "private": null,
+    "tags": null,
+    "license": null,
+    "pretrained_path": null,
+    "chunk_size": 100,
+    "n_action_steps": 100,
+    "normalization_mapping": {
+        "VISUAL": "MEAN_STD",
+        "STATE": "MEAN_STD",
+        "ACTION": "MEAN_STD"
+    },
+    "vision_backbone": "resnet18",
+    "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
+    "replace_final_stride_with_dilation": false,
+    "pre_norm": false,
+    "dim_model": 512,
+    "n_heads": 8,
+    "dim_feedforward": 3200,
+    "feedforward_activation": "relu",
+    "n_encoder_layers": 4,
+    "n_decoder_layers": 1,
+    "use_vae": true,
+    "latent_dim": 32,
+    "n_vae_encoder_layers": 4,
+    "temporal_ensemble_coeff": null,
+    "dropout": 0.1,
+    "kl_weight": 10.0,
+    "optimizer_lr": 1e-05,
+    "optimizer_weight_decay": 0.0001,
+    "optimizer_lr_backbone": 1e-05
+}

checkpoint_035000/model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:59fe0cfd16427f10710b3474442250818c951bdcdffa652dafa7a6a0b0fe5c29
+size 206699736