diff --git a/checkpoint_005000/config.json b/checkpoint_005000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_005000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_005000/model.safetensors b/checkpoint_005000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..dec1c5f14c88aa58743cfb03f02e5fef34d4f74e --- /dev/null +++ b/checkpoint_005000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fe467767d779e0796a2d62521a302d1dc4415bebf18a648666e72ff4f5559006 +size 206699736 diff --git a/checkpoint_005000/policy_postprocessor.json b/checkpoint_005000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_005000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_005000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_005000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_005000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_005000/policy_preprocessor.json b/checkpoint_005000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_005000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_005000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_005000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_005000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_005000/training_metadata.json b/checkpoint_005000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_005000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_005000/training_state.pt b/checkpoint_005000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..514dafa25b627b26310d4a08889c6a635381609e --- /dev/null +++ b/checkpoint_005000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:451d288e35679d0f1a0e55dd286dd942f607219daa718c51a92fede08d8d49cc +size 412908960 diff --git a/checkpoint_010000/config.json b/checkpoint_010000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_010000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_010000/model.safetensors b/checkpoint_010000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..eed94815f13a9124c057f41da4e71fe68b8dea1e --- /dev/null +++ b/checkpoint_010000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:3aab334cc727bc4559d3ff33ca681a6b3abec9176de986f2f98794814d4a8075 +size 206699736 diff --git a/checkpoint_010000/policy_postprocessor.json b/checkpoint_010000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_010000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_010000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_010000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_010000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_010000/policy_preprocessor.json b/checkpoint_010000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_010000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_010000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_010000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_010000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_010000/training_metadata.json b/checkpoint_010000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_010000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_010000/training_state.pt b/checkpoint_010000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..6ab5a58af2c2c5144b432e8d8b250b03a0e9c0d1 --- /dev/null +++ b/checkpoint_010000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:044b49c8af62a7b508d4090e8e3b783f748658fbe217ad7bf2169d26dd4310f6 +size 412908960 diff --git a/checkpoint_015000/config.json b/checkpoint_015000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_015000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_015000/model.safetensors b/checkpoint_015000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..a1a324701d33ef30480cda8f8516ef0f757cf280 --- /dev/null +++ b/checkpoint_015000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e23e76bab6e785709da0faaa23a0d0e391b9f7af83763a90c559f390f373367f +size 206699736 diff --git a/checkpoint_015000/policy_postprocessor.json b/checkpoint_015000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_015000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_015000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_015000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_015000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_015000/policy_preprocessor.json b/checkpoint_015000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_015000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_015000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_015000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_015000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_015000/training_metadata.json b/checkpoint_015000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_015000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_015000/training_state.pt b/checkpoint_015000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..8c6779ae743ff9d4654ebe9f8475e1552473dde4 --- /dev/null +++ b/checkpoint_015000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:94a002e605540c0b70f77dbc19ab021ff3ae85a4705a6fcf226c091d9392b2fb +size 412908960 diff --git a/checkpoint_020000/config.json b/checkpoint_020000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_020000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_020000/model.safetensors b/checkpoint_020000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..dae0dc5e7c3cc6849b8e758f8120fef49d6b4abe --- /dev/null +++ b/checkpoint_020000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1b262f8ff7830a1b095f47eba20298f53d9aa28d89a010f98c2131c49fefce31 +size 206699736 diff --git a/checkpoint_020000/policy_postprocessor.json b/checkpoint_020000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_020000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_020000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_020000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_020000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_020000/policy_preprocessor.json b/checkpoint_020000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_020000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_020000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_020000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_020000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_020000/training_metadata.json b/checkpoint_020000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_020000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_020000/training_state.pt b/checkpoint_020000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..23e3d66ef6618e99608917b89da81bf694c3d9a0 --- /dev/null +++ b/checkpoint_020000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:a934486728f662762a64bc64a904c6c948e3c3c803f4d66630c8e3cfc80b6a9d +size 412908960 diff --git a/checkpoint_025000/config.json b/checkpoint_025000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_025000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_025000/model.safetensors b/checkpoint_025000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..195147f1b011f89204247b3677f28866214a83ec --- /dev/null +++ b/checkpoint_025000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0cf6dec147776dfcfdcfe651f46ec9bfa4f02c3fa4a704fd9e1d29af62556b3 +size 206699736 diff --git a/checkpoint_025000/policy_postprocessor.json b/checkpoint_025000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_025000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_025000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_025000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_025000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_025000/policy_preprocessor.json b/checkpoint_025000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_025000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_025000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_025000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_025000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_025000/training_metadata.json b/checkpoint_025000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_025000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_025000/training_state.pt b/checkpoint_025000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..cc783112188d4a69a30a4e2e3668c74758d47923 --- /dev/null +++ b/checkpoint_025000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1e4ec5f3d9db6284a358f1a77f9d3cd12d80fb16fbcd5073d524a52457d86185 +size 412908960 diff --git a/checkpoint_030000/config.json b/checkpoint_030000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_030000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_030000/model.safetensors b/checkpoint_030000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..3df224efe6b2408490d1622ee3eb46548a3507d5 --- /dev/null +++ b/checkpoint_030000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:68fe1068deae344e65538bb232ca6f3f11218954eba14dd02e145e446a8cfb0d +size 206699736 diff --git a/checkpoint_030000/policy_postprocessor.json b/checkpoint_030000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_030000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_030000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_030000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_030000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_030000/policy_preprocessor.json b/checkpoint_030000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_030000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_030000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_030000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_030000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_030000/training_metadata.json b/checkpoint_030000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_030000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_030000/training_state.pt b/checkpoint_030000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..2e363636b2bbd0961bc618f360cdaf9342101e53 --- /dev/null +++ b/checkpoint_030000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:423456b20e2e9f834fd99e1b53ab74943fb07956960907ea12232d69de9bba78 +size 412908960 diff --git a/checkpoint_035000/config.json b/checkpoint_035000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_035000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_035000/model.safetensors b/checkpoint_035000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..ea3b774ba301b352e00b548faee9505fa36f8bc3 --- /dev/null +++ b/checkpoint_035000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0bdc432c4d22a11c5530a7975a6bb9948900403315e6a4c6b4530f785014ac00 +size 206699736 diff --git a/checkpoint_035000/policy_postprocessor.json b/checkpoint_035000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_035000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_035000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_035000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_035000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_035000/policy_preprocessor.json b/checkpoint_035000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_035000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_035000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_035000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_035000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_035000/training_metadata.json b/checkpoint_035000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_035000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_035000/training_state.pt b/checkpoint_035000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..d0113fef0bbd101057a548e8ccc54ca11f235c4e --- /dev/null +++ b/checkpoint_035000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b424b987c0115b5ae9603ae6992e9826269e033723c40a15aa11e822a79c8099 +size 412908960 diff --git a/checkpoint_040000/config.json b/checkpoint_040000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_040000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_040000/model.safetensors b/checkpoint_040000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..5f99e37bb70a022c603053fe17687b05c95881c9 --- /dev/null +++ b/checkpoint_040000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b76271fd16e4af861c7a56f04c887c974770582fff7f1370aa6f9759c9696b86 +size 206699736 diff --git a/checkpoint_040000/policy_postprocessor.json b/checkpoint_040000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_040000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_040000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_040000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_040000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_040000/policy_preprocessor.json b/checkpoint_040000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_040000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_040000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_040000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_040000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_040000/training_metadata.json b/checkpoint_040000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_040000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_040000/training_state.pt b/checkpoint_040000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..139b795b502a7ee383804fe1da049b6a325ba479 --- /dev/null +++ b/checkpoint_040000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e0872aaeddc7dd1b8d95e33442f82b83a73243f6ae36d72123c067d3a50385da +size 412908960 diff --git a/checkpoint_045000/config.json b/checkpoint_045000/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/checkpoint_045000/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/checkpoint_045000/model.safetensors b/checkpoint_045000/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..be5d879da1d89460b119abc4bdf5428922f26c1d --- /dev/null +++ b/checkpoint_045000/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0bf3fb11138b1250719c1756c24555d2af8d8a79077b0a8f4eccb6afb2964997 +size 206699736 diff --git a/checkpoint_045000/policy_postprocessor.json b/checkpoint_045000/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/checkpoint_045000/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/checkpoint_045000/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoint_045000/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_045000/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_045000/policy_preprocessor.json b/checkpoint_045000/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/checkpoint_045000/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/checkpoint_045000/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoint_045000/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/checkpoint_045000/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/checkpoint_045000/training_metadata.json b/checkpoint_045000/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/checkpoint_045000/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/checkpoint_045000/training_state.pt b/checkpoint_045000/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..a10d1eb0ce15f31e236f99310d73b27bf912933f --- /dev/null +++ b/checkpoint_045000/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d9a3ba143c3e2bc2dce4427391a191d360711f6ce0c7ff39cd49908f3cbde387 +size 412908960 diff --git a/final/config.json b/final/config.json new file mode 100644 index 0000000000000000000000000000000000000000..6c835687da723dcca6ea30285a36badce94914d8 --- /dev/null +++ b/final/config.json @@ -0,0 +1,70 @@ +{ + "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 + ] + } + }, + "output_features": { + "action": { + "type": "ACTION", + "shape": [ + 6 + ] + } + }, + "device": "cuda", + "use_amp": 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 +} \ No newline at end of file diff --git a/final/model.safetensors b/final/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..be5d879da1d89460b119abc4bdf5428922f26c1d --- /dev/null +++ b/final/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0bf3fb11138b1250719c1756c24555d2af8d8a79077b0a8f4eccb6afb2964997 +size 206699736 diff --git a/final/policy_postprocessor.json b/final/policy_postprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..3ba4ac02de62bbf10dd3ff95f2bac8c697e95340 --- /dev/null +++ b/final/policy_postprocessor.json @@ -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 + } + } + ] +} \ No newline at end of file diff --git a/final/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/final/policy_postprocessor_step_0_unnormalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/final/policy_postprocessor_step_0_unnormalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/final/policy_preprocessor.json b/final/policy_preprocessor.json new file mode 100644 index 0000000000000000000000000000000000000000..1a7167e7fef59f8e032de6b34a11e200451c8a0d --- /dev/null +++ b/final/policy_preprocessor.json @@ -0,0 +1,64 @@ +{ + "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 + ] + }, + "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" + } + ] +} \ No newline at end of file diff --git a/final/policy_preprocessor_step_3_normalizer_processor.safetensors b/final/policy_preprocessor_step_3_normalizer_processor.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..efb8d0534b6e0d950bd0a05e0e9ba1d7e5694f68 --- /dev/null +++ b/final/policy_preprocessor_step_3_normalizer_processor.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c71fb5729c86e7035beffea7a07fd91226bed9646b82ad9ed148759abf20f0c7 +size 768 diff --git a/final/training_metadata.json b/final/training_metadata.json new file mode 100644 index 0000000000000000000000000000000000000000..759440cd94bd541e311e1fc335adb6867b5c9839 --- /dev/null +++ b/final/training_metadata.json @@ -0,0 +1,35 @@ +{ + "dataset_repo_id": "danbhf/sim_pick_place_pos2_100ep", + "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": 13700 +} \ No newline at end of file diff --git a/final/training_state.pt b/final/training_state.pt new file mode 100644 index 0000000000000000000000000000000000000000..a10d1eb0ce15f31e236f99310d73b27bf912933f --- /dev/null +++ b/final/training_state.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d9a3ba143c3e2bc2dce4427391a191d360711f6ce0c7ff39cd49908f3cbde387 +size 412908960