diff --git a/checkpoints/001000/pretrained_model/config.json b/checkpoints/001000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/001000/pretrained_model/model.safetensors b/checkpoints/001000/pretrained_model/model.safetensors deleted file mode 100644 index 057253a42707e13e9143a29b3971c3604c855353..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:0671762f65c5267995e5bb92b30e9b45ba845cbf7193ee54383ac0b9ca127ecc -size 206372048 diff --git a/checkpoints/001000/pretrained_model/policy_postprocessor.json b/checkpoints/001000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/001000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/001000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/001000/pretrained_model/policy_preprocessor.json b/checkpoints/001000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/001000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/001000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/001000/pretrained_model/train_config.json b/checkpoints/001000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/001000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/001000/training_state/optimizer_param_groups.json b/checkpoints/001000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/001000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/001000/training_state/optimizer_state.safetensors b/checkpoints/001000/training_state/optimizer_state.safetensors deleted file mode 100644 index f600e2826524849f77129dac097cf9622662cf21..0000000000000000000000000000000000000000 --- a/checkpoints/001000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:75fb596370d8bc44d34ae415fabf738ac53a1c229826a4a70bf25288fa72fc81 -size 412489972 diff --git a/checkpoints/001000/training_state/rng_state.safetensors b/checkpoints/001000/training_state/rng_state.safetensors deleted file mode 100644 index 6ff4ac32767bf993a46227661731ac7c65b2fc0a..0000000000000000000000000000000000000000 --- a/checkpoints/001000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:32b149efbbcead531fdc3eb8b173ed9c054e2f39d66b043b45acf35bbf81644c -size 15708 diff --git a/checkpoints/001000/training_state/training_step.json b/checkpoints/001000/training_state/training_step.json deleted file mode 100644 index d98c94f5b78238bf495ac68b9f9fb446cfac5c07..0000000000000000000000000000000000000000 --- a/checkpoints/001000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 1000 -} \ No newline at end of file diff --git a/checkpoints/002000/pretrained_model/config.json b/checkpoints/002000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/002000/pretrained_model/model.safetensors b/checkpoints/002000/pretrained_model/model.safetensors deleted file mode 100644 index fffac99f7c9a88967d3bc0f22b978525efd3563f..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:ac0edc11e5dc40b800f94fdf9a5576196b49dcd23f9b40c39e44141c3e0138ce -size 206372048 diff --git a/checkpoints/002000/pretrained_model/policy_postprocessor.json b/checkpoints/002000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/002000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/002000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/002000/pretrained_model/policy_preprocessor.json b/checkpoints/002000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/002000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/002000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/002000/pretrained_model/train_config.json b/checkpoints/002000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/002000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/002000/training_state/optimizer_param_groups.json b/checkpoints/002000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/002000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/002000/training_state/optimizer_state.safetensors b/checkpoints/002000/training_state/optimizer_state.safetensors deleted file mode 100644 index 7675bea93acbc43d0bfd15d24d46f767169a4436..0000000000000000000000000000000000000000 --- a/checkpoints/002000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a8c09c6644e57234925e8d017c6fba7e954b61138b98bfff510d99dc866d73b4 -size 412489972 diff --git a/checkpoints/002000/training_state/rng_state.safetensors b/checkpoints/002000/training_state/rng_state.safetensors deleted file mode 100644 index bc3df3d591805bc4b42e03b5018325ece5e3f903..0000000000000000000000000000000000000000 --- a/checkpoints/002000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:24d1e3d5257e149c82bd9c4a97f3329c147c5cd57e15bd010dee2fd429034f69 -size 15708 diff --git a/checkpoints/002000/training_state/training_step.json b/checkpoints/002000/training_state/training_step.json deleted file mode 100644 index 39090bbb986edb821e1602990d19357dcdb5d2ae..0000000000000000000000000000000000000000 --- a/checkpoints/002000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 2000 -} \ No newline at end of file diff --git a/checkpoints/003000/pretrained_model/config.json b/checkpoints/003000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/003000/pretrained_model/model.safetensors b/checkpoints/003000/pretrained_model/model.safetensors deleted file mode 100644 index 792f65513a1c33351cae85de184e5b08e572a94c..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:05f410bfc7290baf61f059ddbf74b556eef75556b4236e811f5858b696f1fd30 -size 206372048 diff --git a/checkpoints/003000/pretrained_model/policy_postprocessor.json b/checkpoints/003000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/003000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/003000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/003000/pretrained_model/policy_preprocessor.json b/checkpoints/003000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/003000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/003000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/003000/pretrained_model/train_config.json b/checkpoints/003000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/003000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/003000/training_state/optimizer_param_groups.json b/checkpoints/003000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/003000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/003000/training_state/optimizer_state.safetensors b/checkpoints/003000/training_state/optimizer_state.safetensors deleted file mode 100644 index 85eff5e1421c88c5a51fb2b6fa6d67a968d66751..0000000000000000000000000000000000000000 --- a/checkpoints/003000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:eeac2d05299bd63b3aa53390f975a4545840ae41929afb77f8b52edb0b681b01 -size 412489972 diff --git a/checkpoints/003000/training_state/rng_state.safetensors b/checkpoints/003000/training_state/rng_state.safetensors deleted file mode 100644 index e7bb87ebc6b670342c28818ce87670af5d2a13ff..0000000000000000000000000000000000000000 --- a/checkpoints/003000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:9339045e8c5a1c161ad035b22fe382c61aabc0d92d7cd8c2ea316ce1781baad5 -size 15708 diff --git a/checkpoints/003000/training_state/training_step.json b/checkpoints/003000/training_state/training_step.json deleted file mode 100644 index c1a44127b7dfea653fd776d529fa83c55d32081c..0000000000000000000000000000000000000000 --- a/checkpoints/003000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 3000 -} \ No newline at end of file diff --git a/checkpoints/004000/pretrained_model/config.json b/checkpoints/004000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/004000/pretrained_model/model.safetensors b/checkpoints/004000/pretrained_model/model.safetensors deleted file mode 100644 index 91ca79417c48eaa7a0da6e3ee8f347900c45c6c2..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:3345e3afdcc56b935478d83da62732311e103b9a1796a75cb2e1ec774d18f6fa -size 206372048 diff --git a/checkpoints/004000/pretrained_model/policy_postprocessor.json b/checkpoints/004000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/004000/pretrained_model/policy_preprocessor.json b/checkpoints/004000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/004000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/004000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/004000/pretrained_model/train_config.json b/checkpoints/004000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/004000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/004000/training_state/optimizer_param_groups.json b/checkpoints/004000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/004000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/004000/training_state/optimizer_state.safetensors b/checkpoints/004000/training_state/optimizer_state.safetensors deleted file mode 100644 index eec346582b448523850b7852c2b2b4e059ed28c8..0000000000000000000000000000000000000000 --- a/checkpoints/004000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:87bd1e0b23e6b3ec0781ec7954f2e2deb5af4032749e99e7f34d150675da9fab -size 412489972 diff --git a/checkpoints/004000/training_state/rng_state.safetensors b/checkpoints/004000/training_state/rng_state.safetensors deleted file mode 100644 index 606672aac40654f683b43526b0cfded261b5b019..0000000000000000000000000000000000000000 --- a/checkpoints/004000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:91cbfa44989f1c76c88d2614fbfd651e38bb9fc3ab132381f0b749266f796e4d -size 15708 diff --git a/checkpoints/004000/training_state/training_step.json b/checkpoints/004000/training_state/training_step.json deleted file mode 100644 index 75d870521c191f77dd9eaa4d83486eab6e768f69..0000000000000000000000000000000000000000 --- a/checkpoints/004000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 4000 -} \ No newline at end of file diff --git a/checkpoints/006000/pretrained_model/config.json b/checkpoints/006000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/006000/pretrained_model/model.safetensors b/checkpoints/006000/pretrained_model/model.safetensors deleted file mode 100644 index e5753aadc8f091edc59d15cbac41016e8dc26383..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a7a1253cd66d29321c1b4de0d89ebb0b234feb5815714c0f2fd81855027e7e13 -size 206372048 diff --git a/checkpoints/006000/pretrained_model/policy_postprocessor.json b/checkpoints/006000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/006000/pretrained_model/policy_preprocessor.json b/checkpoints/006000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/006000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/006000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/006000/pretrained_model/train_config.json b/checkpoints/006000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/006000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/006000/training_state/optimizer_param_groups.json b/checkpoints/006000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/006000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/006000/training_state/optimizer_state.safetensors b/checkpoints/006000/training_state/optimizer_state.safetensors deleted file mode 100644 index d955f508a426e341047bd161ffaac1ec7295baa1..0000000000000000000000000000000000000000 --- a/checkpoints/006000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:5fc2ee228ba3c296d3a47103615c76e7b1e47253bb765c2eaa522dcc5136ce34 -size 412489972 diff --git a/checkpoints/006000/training_state/rng_state.safetensors b/checkpoints/006000/training_state/rng_state.safetensors deleted file mode 100644 index dc04c5b3013a77b9e6fa62a17e99840d9782d492..0000000000000000000000000000000000000000 --- a/checkpoints/006000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:9f16c85eed6bc3fa85a6f8fbebe0a4f9eeef52cda990d73c5b2dd18495dee2ae -size 15708 diff --git a/checkpoints/006000/training_state/training_step.json b/checkpoints/006000/training_state/training_step.json deleted file mode 100644 index e267ac589be64705f8674638b9f5099c886778da..0000000000000000000000000000000000000000 --- a/checkpoints/006000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 6000 -} \ No newline at end of file diff --git a/checkpoints/007000/pretrained_model/config.json b/checkpoints/007000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/007000/pretrained_model/model.safetensors b/checkpoints/007000/pretrained_model/model.safetensors deleted file mode 100644 index 6637ff3f6fe24f4abadee7d70eb777c0f35d2df1..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:66252f7e5a53b7dd9201b23b61dfd79a067f8f6026d8ff44f6527bd7530db900 -size 206372048 diff --git a/checkpoints/007000/pretrained_model/policy_postprocessor.json b/checkpoints/007000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/007000/pretrained_model/policy_preprocessor.json b/checkpoints/007000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/007000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/007000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/007000/pretrained_model/train_config.json b/checkpoints/007000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/007000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/007000/training_state/optimizer_param_groups.json b/checkpoints/007000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/007000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/007000/training_state/optimizer_state.safetensors b/checkpoints/007000/training_state/optimizer_state.safetensors deleted file mode 100644 index 15f084794f7bfc694bca1150a70d2fc4459de929..0000000000000000000000000000000000000000 --- a/checkpoints/007000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f969824d25fd56448ac27dda10978f9b16215bdf6c23295a7dfd307460d5713b -size 412489972 diff --git a/checkpoints/007000/training_state/rng_state.safetensors b/checkpoints/007000/training_state/rng_state.safetensors deleted file mode 100644 index d570d78a5e043380dd0606e0de3077061a193dd4..0000000000000000000000000000000000000000 --- a/checkpoints/007000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:488e3454212a3de19a4c0fc57d9c28b76860bd99121b5289fa20dfa929dafdd4 -size 15708 diff --git a/checkpoints/007000/training_state/training_step.json b/checkpoints/007000/training_state/training_step.json deleted file mode 100644 index 1f4c730c4d88ece076ae9d4e01ed7fbe60b0415d..0000000000000000000000000000000000000000 --- a/checkpoints/007000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 7000 -} \ No newline at end of file diff --git a/checkpoints/008000/pretrained_model/config.json b/checkpoints/008000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/008000/pretrained_model/model.safetensors b/checkpoints/008000/pretrained_model/model.safetensors deleted file mode 100644 index 3a370cb818f49877a358622dbab6ef591cba0bbd..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:04631a0d36cfe58f2f1b89e0f7da61313f718fd5a81d2333c016567eb920c945 -size 206372048 diff --git a/checkpoints/008000/pretrained_model/policy_postprocessor.json b/checkpoints/008000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/008000/pretrained_model/policy_preprocessor.json b/checkpoints/008000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/008000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/008000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/008000/pretrained_model/train_config.json b/checkpoints/008000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/008000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/008000/training_state/optimizer_param_groups.json b/checkpoints/008000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/008000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/008000/training_state/optimizer_state.safetensors b/checkpoints/008000/training_state/optimizer_state.safetensors deleted file mode 100644 index 72aa750fdab80d5738e1f6a8f728b2e59bf64626..0000000000000000000000000000000000000000 --- a/checkpoints/008000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:4601839cd1ea20aac419f14c589032a0af5d6b430ce12e5c60a46f30f814effb -size 412489972 diff --git a/checkpoints/008000/training_state/rng_state.safetensors b/checkpoints/008000/training_state/rng_state.safetensors deleted file mode 100644 index d3f313a00f46241e16dfa0b559a229fd5ab03ef7..0000000000000000000000000000000000000000 --- a/checkpoints/008000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:528ee478543caac3fedc72c2c0386ce18e7caa06a21635051565c7f1830edb89 -size 15708 diff --git a/checkpoints/008000/training_state/training_step.json b/checkpoints/008000/training_state/training_step.json deleted file mode 100644 index 74d8cba01cab8506617b2cbae6f268fe80fbfa79..0000000000000000000000000000000000000000 --- a/checkpoints/008000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 8000 -} \ No newline at end of file diff --git a/checkpoints/009000/pretrained_model/config.json b/checkpoints/009000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/009000/pretrained_model/model.safetensors b/checkpoints/009000/pretrained_model/model.safetensors deleted file mode 100644 index e33a24e5c78d161d76074b436080e850d1777072..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:b528d04432efa01a476ebbd5729f8c512826d59ffa2645e2f31c9be9a9719441 -size 206372048 diff --git a/checkpoints/009000/pretrained_model/policy_postprocessor.json b/checkpoints/009000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/009000/pretrained_model/policy_preprocessor.json b/checkpoints/009000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/009000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/009000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/009000/pretrained_model/train_config.json b/checkpoints/009000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/009000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/009000/training_state/optimizer_param_groups.json b/checkpoints/009000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/009000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/009000/training_state/optimizer_state.safetensors b/checkpoints/009000/training_state/optimizer_state.safetensors deleted file mode 100644 index 1d08a7472bd64121e53aab37098a5b7eb3cfe4a3..0000000000000000000000000000000000000000 --- a/checkpoints/009000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:d1210da85ac90b22366efd09218973eecc8d21551c453b0798123054b7234788 -size 412489972 diff --git a/checkpoints/009000/training_state/rng_state.safetensors b/checkpoints/009000/training_state/rng_state.safetensors deleted file mode 100644 index 7fa50e4d3554784c5789d90c0c69e95ce2e42e68..0000000000000000000000000000000000000000 --- a/checkpoints/009000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:fe35a4ac72c16255671e41ae2224f7ec7b7bedc4a3e0e84ac068519a4ca22236 -size 15708 diff --git a/checkpoints/009000/training_state/training_step.json b/checkpoints/009000/training_state/training_step.json deleted file mode 100644 index 7247596175281e9d3f295d2b6936944a71b64d34..0000000000000000000000000000000000000000 --- a/checkpoints/009000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 9000 -} \ No newline at end of file diff --git a/checkpoints/011000/pretrained_model/config.json b/checkpoints/011000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/011000/pretrained_model/model.safetensors b/checkpoints/011000/pretrained_model/model.safetensors deleted file mode 100644 index a7b96c64d49c5911be52a9771fd481645eb0dabe..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:0a0147ec4060fec10432c0369d4f890134c02079e51964adc9ccb9531648352d -size 206372048 diff --git a/checkpoints/011000/pretrained_model/policy_postprocessor.json b/checkpoints/011000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/011000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/011000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/011000/pretrained_model/policy_preprocessor.json b/checkpoints/011000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/011000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/011000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/011000/pretrained_model/train_config.json b/checkpoints/011000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/011000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/011000/training_state/optimizer_param_groups.json b/checkpoints/011000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/011000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/011000/training_state/optimizer_state.safetensors b/checkpoints/011000/training_state/optimizer_state.safetensors deleted file mode 100644 index 16a1f2242b382d07b3fcea02350c639aac7a0eb2..0000000000000000000000000000000000000000 --- a/checkpoints/011000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:495689f34013c620b2d3a542a751cff8d86f60979a21206d523f2538ac8c788d -size 412489972 diff --git a/checkpoints/011000/training_state/rng_state.safetensors b/checkpoints/011000/training_state/rng_state.safetensors deleted file mode 100644 index 27a6b2c1ce11818ce8b62ee96e5fabeb5d59d150..0000000000000000000000000000000000000000 --- a/checkpoints/011000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:cd7611f5f37ef5fcb77dcb39907f6f43d7d3bffec8617b7f9fb102cb081fee3e -size 15708 diff --git a/checkpoints/011000/training_state/training_step.json b/checkpoints/011000/training_state/training_step.json deleted file mode 100644 index 2389cf08c9a887d74ab54bbe783d6a4eccbbee72..0000000000000000000000000000000000000000 --- a/checkpoints/011000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 11000 -} \ No newline at end of file diff --git a/checkpoints/012000/pretrained_model/config.json b/checkpoints/012000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/012000/pretrained_model/model.safetensors b/checkpoints/012000/pretrained_model/model.safetensors deleted file mode 100644 index 76b39b6a0bf6392f791b3d856e60a873728c3cb7..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:205a5fb7295e8f101803bb5bc8206bc6db3ddc25557d955c7b33085e82b1d0bd -size 206372048 diff --git a/checkpoints/012000/pretrained_model/policy_postprocessor.json b/checkpoints/012000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/012000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/012000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/012000/pretrained_model/policy_preprocessor.json b/checkpoints/012000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/012000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/012000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/012000/pretrained_model/train_config.json b/checkpoints/012000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/012000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/012000/training_state/optimizer_param_groups.json b/checkpoints/012000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/012000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/012000/training_state/optimizer_state.safetensors b/checkpoints/012000/training_state/optimizer_state.safetensors deleted file mode 100644 index a262a86fbe0da4bbeebf8142aedf0a34a2e2f85f..0000000000000000000000000000000000000000 --- a/checkpoints/012000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2f8a08321ccc658d718e077779bbd6c0e977a1c09ef87ed4aa6d497ec21b2aa1 -size 412489972 diff --git a/checkpoints/012000/training_state/rng_state.safetensors b/checkpoints/012000/training_state/rng_state.safetensors deleted file mode 100644 index a9858b1cc8ced6ca32fb8146504054eecccbacab..0000000000000000000000000000000000000000 --- a/checkpoints/012000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:e7da6ff00eee78bdb3526ee11811c1fbe99bb3bc6a433db383f452ec2aa9a0bb -size 15708 diff --git a/checkpoints/012000/training_state/training_step.json b/checkpoints/012000/training_state/training_step.json deleted file mode 100644 index c4fb27ac819b81943e6545c7c18510bdfb8eae1b..0000000000000000000000000000000000000000 --- a/checkpoints/012000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 12000 -} \ No newline at end of file diff --git a/checkpoints/013000/pretrained_model/config.json b/checkpoints/013000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/013000/pretrained_model/model.safetensors b/checkpoints/013000/pretrained_model/model.safetensors deleted file mode 100644 index 0cadc9e3403322b56985e5889d2ab11376a5e1ea..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:4402f22810c01374fefd49bb7fab2d39b6009ee9db74d23de36d55c529c02aa3 -size 206372048 diff --git a/checkpoints/013000/pretrained_model/policy_postprocessor.json b/checkpoints/013000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/013000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/013000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/013000/pretrained_model/policy_preprocessor.json b/checkpoints/013000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/013000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/013000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/013000/pretrained_model/train_config.json b/checkpoints/013000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/013000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/013000/training_state/optimizer_param_groups.json b/checkpoints/013000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/013000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/013000/training_state/optimizer_state.safetensors b/checkpoints/013000/training_state/optimizer_state.safetensors deleted file mode 100644 index 8bd8bf64f91204b9b3066e230fe991f9246f78f5..0000000000000000000000000000000000000000 --- a/checkpoints/013000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:c593f187538aead5c23c0d4fbfc107c24049015504142359fb8fa9e1a1122c36 -size 412489972 diff --git a/checkpoints/013000/training_state/rng_state.safetensors b/checkpoints/013000/training_state/rng_state.safetensors deleted file mode 100644 index e5d4ae57d9a2653cf096893053c41d13ae384cbb..0000000000000000000000000000000000000000 --- a/checkpoints/013000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a10458ed49c6a23ad98f95bf62105ee4fcb9e4bcb82e750c9adc9101c50bf2b4 -size 15708 diff --git a/checkpoints/013000/training_state/training_step.json b/checkpoints/013000/training_state/training_step.json deleted file mode 100644 index 6f5915e7fb38c851f5b48d58c1f66fe9c9d147ed..0000000000000000000000000000000000000000 --- a/checkpoints/013000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 13000 -} \ No newline at end of file diff --git a/checkpoints/014000/pretrained_model/config.json b/checkpoints/014000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/014000/pretrained_model/model.safetensors b/checkpoints/014000/pretrained_model/model.safetensors deleted file mode 100644 index b77112833cac96f570f8217195aa1fdb31daa203..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:58de2147d9224c4e0431ab0f2afa61d47a29e34e42b70ba1e1dc134ea0d7d2e6 -size 206372048 diff --git a/checkpoints/014000/pretrained_model/policy_postprocessor.json b/checkpoints/014000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/014000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/014000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/014000/pretrained_model/policy_preprocessor.json b/checkpoints/014000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/014000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/014000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/014000/pretrained_model/train_config.json b/checkpoints/014000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/014000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/014000/training_state/optimizer_param_groups.json b/checkpoints/014000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/014000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/014000/training_state/optimizer_state.safetensors b/checkpoints/014000/training_state/optimizer_state.safetensors deleted file mode 100644 index 29c0621e229da7c4b2b3d66dcd5da4f1391245ad..0000000000000000000000000000000000000000 --- a/checkpoints/014000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:c20ad60535bf4686f9dcb51929867e6b62ac5f9fe2a6597add6b99afa7345c65 -size 412489972 diff --git a/checkpoints/014000/training_state/rng_state.safetensors b/checkpoints/014000/training_state/rng_state.safetensors deleted file mode 100644 index c0c99a0fb8427a7c82f7146b8181a2f1554e3dd9..0000000000000000000000000000000000000000 --- a/checkpoints/014000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:cf9c67afab05769053dcab1e126cf033e35df94b073fd5cde247ef2eb13e4bbd -size 15708 diff --git a/checkpoints/014000/training_state/training_step.json b/checkpoints/014000/training_state/training_step.json deleted file mode 100644 index 560b82990f5c806d547ed82a5432ba60aec285e1..0000000000000000000000000000000000000000 --- a/checkpoints/014000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 14000 -} \ No newline at end of file diff --git a/checkpoints/016000/pretrained_model/config.json b/checkpoints/016000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/016000/pretrained_model/model.safetensors b/checkpoints/016000/pretrained_model/model.safetensors deleted file mode 100644 index 5a23368186202c0d04a4ef4fa0fc4a92f8b30839..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:709119c262b63088ba1c2ffb02f6a644d95411482a7a86c40116f8f8e5f80b92 -size 206372048 diff --git a/checkpoints/016000/pretrained_model/policy_postprocessor.json b/checkpoints/016000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/016000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/016000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/016000/pretrained_model/policy_preprocessor.json b/checkpoints/016000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/016000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/016000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/016000/pretrained_model/train_config.json b/checkpoints/016000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/016000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/016000/training_state/optimizer_param_groups.json b/checkpoints/016000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/016000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/016000/training_state/optimizer_state.safetensors b/checkpoints/016000/training_state/optimizer_state.safetensors deleted file mode 100644 index d519c34b9ac7ec440d02eaf066827339d2459966..0000000000000000000000000000000000000000 --- a/checkpoints/016000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:3ebefbd8c4023f717ab9cb7f9210e50f0a6ba4acfce9015636837bb471b31dd0 -size 412489972 diff --git a/checkpoints/016000/training_state/rng_state.safetensors b/checkpoints/016000/training_state/rng_state.safetensors deleted file mode 100644 index e030e3d52ebac89bb5739dcb54f2a187419a2ab8..0000000000000000000000000000000000000000 --- a/checkpoints/016000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:194bc9f27f39dada5094360be8449c29284341a49dcc19c60da4155cafea8898 -size 15708 diff --git a/checkpoints/016000/training_state/training_step.json b/checkpoints/016000/training_state/training_step.json deleted file mode 100644 index 798020d90d4c39414dfab88708ecefd52f2e9ac5..0000000000000000000000000000000000000000 --- a/checkpoints/016000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 16000 -} \ No newline at end of file diff --git a/checkpoints/017000/pretrained_model/config.json b/checkpoints/017000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/017000/pretrained_model/model.safetensors b/checkpoints/017000/pretrained_model/model.safetensors deleted file mode 100644 index 182ab1cdb986335c487f7cf927786659b30e6e80..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:855a65c7c85dd608e429d2f498585ff61465806946ebf0b310047af5a9ee67f5 -size 206372048 diff --git a/checkpoints/017000/pretrained_model/policy_postprocessor.json b/checkpoints/017000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/017000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/017000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/017000/pretrained_model/policy_preprocessor.json b/checkpoints/017000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/017000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/017000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/017000/pretrained_model/train_config.json b/checkpoints/017000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/017000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/017000/training_state/optimizer_param_groups.json b/checkpoints/017000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/017000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/017000/training_state/optimizer_state.safetensors b/checkpoints/017000/training_state/optimizer_state.safetensors deleted file mode 100644 index a6a1c53b852a21bfb5aebd46f176283c03a38aee..0000000000000000000000000000000000000000 --- a/checkpoints/017000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:6e54f652480f516520360a57c71b965d69d5b0a396a2b494281bffc19faee338 -size 412489972 diff --git a/checkpoints/017000/training_state/rng_state.safetensors b/checkpoints/017000/training_state/rng_state.safetensors deleted file mode 100644 index 8e1522c6eef1e18436366ac3a3a3285c54bb99d1..0000000000000000000000000000000000000000 --- a/checkpoints/017000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:98bc7ed62c889a6f1f5ab28a0f048af43f09dcaa5fb78e15bff00bac85e22a32 -size 15708 diff --git a/checkpoints/017000/training_state/training_step.json b/checkpoints/017000/training_state/training_step.json deleted file mode 100644 index c397907165bb49af5c8ed9f0802e1ca19fad31fb..0000000000000000000000000000000000000000 --- a/checkpoints/017000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 17000 -} \ No newline at end of file diff --git a/checkpoints/018000/pretrained_model/config.json b/checkpoints/018000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/018000/pretrained_model/model.safetensors b/checkpoints/018000/pretrained_model/model.safetensors deleted file mode 100644 index 078649e76a88d18453547746ad2a83e215b5eab0..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:e31928c3e058366509aa35164449023989252b06b72fb2d1b10c6cf88d20bbbb -size 206372048 diff --git a/checkpoints/018000/pretrained_model/policy_postprocessor.json b/checkpoints/018000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/018000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/018000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/018000/pretrained_model/policy_preprocessor.json b/checkpoints/018000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/018000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/018000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/018000/pretrained_model/train_config.json b/checkpoints/018000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/018000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/018000/training_state/optimizer_param_groups.json b/checkpoints/018000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/018000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/018000/training_state/optimizer_state.safetensors b/checkpoints/018000/training_state/optimizer_state.safetensors deleted file mode 100644 index b262a4f12555e4c92d4415c0caa53c6b59fc30d8..0000000000000000000000000000000000000000 --- a/checkpoints/018000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7b1472f7df0757bf0957deb9c1218f0df4dd740b644453367f7d0bf250367358 -size 412489972 diff --git a/checkpoints/018000/training_state/rng_state.safetensors b/checkpoints/018000/training_state/rng_state.safetensors deleted file mode 100644 index 0511672877cb652d2cc3fbf2ad4cdfa1c2f91660..0000000000000000000000000000000000000000 --- a/checkpoints/018000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:31a833158832c61ae40220e107df27dba4fb001d8018ee1c0b5f5b297734f4a3 -size 15708 diff --git a/checkpoints/018000/training_state/training_step.json b/checkpoints/018000/training_state/training_step.json deleted file mode 100644 index 332a3c17a399c305da7a515c33ffb382d0aff339..0000000000000000000000000000000000000000 --- a/checkpoints/018000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 18000 -} \ No newline at end of file diff --git a/checkpoints/019000/pretrained_model/config.json b/checkpoints/019000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/019000/pretrained_model/model.safetensors b/checkpoints/019000/pretrained_model/model.safetensors deleted file mode 100644 index aed1386c7e18dbb69f1c78dcf414ed02a69f3350..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f9224f75f9d5f78edc9268f66c006de35838e7fa38144f09bd01c48ca5420ecc -size 206372048 diff --git a/checkpoints/019000/pretrained_model/policy_postprocessor.json b/checkpoints/019000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/019000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/019000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/019000/pretrained_model/policy_preprocessor.json b/checkpoints/019000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/019000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/019000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/019000/pretrained_model/train_config.json b/checkpoints/019000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/019000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/019000/training_state/optimizer_param_groups.json b/checkpoints/019000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/019000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/019000/training_state/optimizer_state.safetensors b/checkpoints/019000/training_state/optimizer_state.safetensors deleted file mode 100644 index 185a1a993efc867e801b9d014f77239c19f73839..0000000000000000000000000000000000000000 --- a/checkpoints/019000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:b0c40b537572d2190f9ed50859a7fb390280faae3a98838499b0c295a2a5baf9 -size 412489972 diff --git a/checkpoints/019000/training_state/rng_state.safetensors b/checkpoints/019000/training_state/rng_state.safetensors deleted file mode 100644 index b5797c86c3f7fcdf81e816b544aac5028267a9c1..0000000000000000000000000000000000000000 --- a/checkpoints/019000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:82ddd9c2a3d21f21f9c664fc3ed06b55a19b3bad95c42da4a525f3f53c5bdf17 -size 15708 diff --git a/checkpoints/019000/training_state/training_step.json b/checkpoints/019000/training_state/training_step.json deleted file mode 100644 index 56ff920c81fcd8f0a4df20161e29926b955a4012..0000000000000000000000000000000000000000 --- a/checkpoints/019000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 19000 -} \ No newline at end of file diff --git a/checkpoints/021000/pretrained_model/config.json b/checkpoints/021000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/021000/pretrained_model/model.safetensors b/checkpoints/021000/pretrained_model/model.safetensors deleted file mode 100644 index 93057307cc9b05a8fbb4c5f90735d3870ea489d0..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:089b0efce46a1ac8d9da21cd9bec9c8b25c19a9793fbd85ffe629b0c39fb4454 -size 206372048 diff --git a/checkpoints/021000/pretrained_model/policy_postprocessor.json b/checkpoints/021000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/021000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/021000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/021000/pretrained_model/policy_preprocessor.json b/checkpoints/021000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/021000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/021000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/021000/pretrained_model/train_config.json b/checkpoints/021000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/021000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/021000/training_state/optimizer_param_groups.json b/checkpoints/021000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/021000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/021000/training_state/optimizer_state.safetensors b/checkpoints/021000/training_state/optimizer_state.safetensors deleted file mode 100644 index 5a7ee5b878094754fa83209beed15bdae087a57b..0000000000000000000000000000000000000000 --- a/checkpoints/021000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:320b4a705b18f756488926f59820ef04824cafd7259435b184958cbd8bd44ca0 -size 412489972 diff --git a/checkpoints/021000/training_state/rng_state.safetensors b/checkpoints/021000/training_state/rng_state.safetensors deleted file mode 100644 index 7bc4f1740a679b0e7d0559a0d044f4922116e73f..0000000000000000000000000000000000000000 --- a/checkpoints/021000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:788601b642569a3172ff8c7dc37832709d6efb13fb77a864097080ee2c278a3f -size 15708 diff --git a/checkpoints/021000/training_state/training_step.json b/checkpoints/021000/training_state/training_step.json deleted file mode 100644 index 7820c04576848a4c4bc63dcb05df4faf1161c7f5..0000000000000000000000000000000000000000 --- a/checkpoints/021000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 21000 -} \ No newline at end of file diff --git a/checkpoints/022000/pretrained_model/config.json b/checkpoints/022000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/022000/pretrained_model/model.safetensors b/checkpoints/022000/pretrained_model/model.safetensors deleted file mode 100644 index e98c1601bd6a1b086cacf190b76378787a7589e5..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7232f5dd04830477cb7b4f06149b4d7621557b6a608e94876dbc1281318f6d6c -size 206372048 diff --git a/checkpoints/022000/pretrained_model/policy_postprocessor.json b/checkpoints/022000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/022000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/022000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/022000/pretrained_model/policy_preprocessor.json b/checkpoints/022000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/022000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/022000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/022000/pretrained_model/train_config.json b/checkpoints/022000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/022000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/022000/training_state/optimizer_param_groups.json b/checkpoints/022000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/022000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/022000/training_state/optimizer_state.safetensors b/checkpoints/022000/training_state/optimizer_state.safetensors deleted file mode 100644 index 1ee871a995fb98b59b809c6f8946cfee824417be..0000000000000000000000000000000000000000 --- a/checkpoints/022000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:0788ca6184d7b8a25f9211001e7efb82bf0944e599c6fa73a60edf6b72b071c5 -size 412489972 diff --git a/checkpoints/022000/training_state/rng_state.safetensors b/checkpoints/022000/training_state/rng_state.safetensors deleted file mode 100644 index c0b4eb0b2e82c6391c9a66cc448dec2c6feda2ad..0000000000000000000000000000000000000000 --- a/checkpoints/022000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:13e50cb0485024011dfe16cc062e1785da1892428e6e7cc5f6585e93127dd8a0 -size 15708 diff --git a/checkpoints/022000/training_state/training_step.json b/checkpoints/022000/training_state/training_step.json deleted file mode 100644 index d60e27dffb5d802ca16598a852aec850971cbb39..0000000000000000000000000000000000000000 --- a/checkpoints/022000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 22000 -} \ No newline at end of file diff --git a/checkpoints/023000/pretrained_model/config.json b/checkpoints/023000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/023000/pretrained_model/model.safetensors b/checkpoints/023000/pretrained_model/model.safetensors deleted file mode 100644 index 90d1b9e76628dda920ea26688c486a14f85f740f..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f83a50ae025b48d9b122628d65cbb3e1c6996a4b056229e6e328cde9329ddac0 -size 206372048 diff --git a/checkpoints/023000/pretrained_model/policy_postprocessor.json b/checkpoints/023000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/023000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/023000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/023000/pretrained_model/policy_preprocessor.json b/checkpoints/023000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/023000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/023000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/023000/pretrained_model/train_config.json b/checkpoints/023000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/023000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/023000/training_state/optimizer_param_groups.json b/checkpoints/023000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/023000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/023000/training_state/optimizer_state.safetensors b/checkpoints/023000/training_state/optimizer_state.safetensors deleted file mode 100644 index 3aeb8c77e5113b8292c010e961e5d74109e817d2..0000000000000000000000000000000000000000 --- a/checkpoints/023000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2c710810ef37b9f00daa18410b36a0c6f25b58b644a1ed9e2dcc746e3e6d3eaf -size 412489972 diff --git a/checkpoints/023000/training_state/rng_state.safetensors b/checkpoints/023000/training_state/rng_state.safetensors deleted file mode 100644 index 8cf190eae6532ae3323735ec5ade245d8d773e9b..0000000000000000000000000000000000000000 --- a/checkpoints/023000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:32fcdb3ac9ec804916672a50444d245178d0d595791f139ccb52d2a904a2735d -size 15708 diff --git a/checkpoints/023000/training_state/training_step.json b/checkpoints/023000/training_state/training_step.json deleted file mode 100644 index e2ae872691c0f4c384a75e3015a6caf86c014370..0000000000000000000000000000000000000000 --- a/checkpoints/023000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 23000 -} \ No newline at end of file diff --git a/checkpoints/024000/pretrained_model/config.json b/checkpoints/024000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/024000/pretrained_model/model.safetensors b/checkpoints/024000/pretrained_model/model.safetensors deleted file mode 100644 index e72d20bb14d432659bae990926f9ad751d45f4af..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:94bb441bdd393d0d05c295aabcbfd369b90450ee6480e0421a08033e24298591 -size 206372048 diff --git a/checkpoints/024000/pretrained_model/policy_postprocessor.json b/checkpoints/024000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/024000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/024000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/024000/pretrained_model/policy_preprocessor.json b/checkpoints/024000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/024000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/024000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/024000/pretrained_model/train_config.json b/checkpoints/024000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/024000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/024000/training_state/optimizer_param_groups.json b/checkpoints/024000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/024000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/024000/training_state/optimizer_state.safetensors b/checkpoints/024000/training_state/optimizer_state.safetensors deleted file mode 100644 index ac6396f91f4755ce34ad9bbaa139b2396da51ffa..0000000000000000000000000000000000000000 --- a/checkpoints/024000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:608334df6c1c9f8a7cf095e30e01cfb818b0dcc9741ba1ef7fc42a531db4d1f6 -size 412489972 diff --git a/checkpoints/024000/training_state/rng_state.safetensors b/checkpoints/024000/training_state/rng_state.safetensors deleted file mode 100644 index 551ec68f2d238d2d3f18be37e6f954e7ad9e4aa3..0000000000000000000000000000000000000000 --- a/checkpoints/024000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a24e29c8ea5e8857916629e02732a59b873f7ce4e978ccda0f1c61d0e6f061c7 -size 15708 diff --git a/checkpoints/024000/training_state/training_step.json b/checkpoints/024000/training_state/training_step.json deleted file mode 100644 index 4bf22a00d4c8b45342d351916c6e2f27ab1dd7d9..0000000000000000000000000000000000000000 --- a/checkpoints/024000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 24000 -} \ No newline at end of file diff --git a/checkpoints/026000/pretrained_model/config.json b/checkpoints/026000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/026000/pretrained_model/model.safetensors b/checkpoints/026000/pretrained_model/model.safetensors deleted file mode 100644 index d560a104a24b7d65d07873f1109bb87ad719bcde..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:edada25ac506f35ddeedada49df63f303dd1c7b99f1698273ec124e88131de94 -size 206372048 diff --git a/checkpoints/026000/pretrained_model/policy_postprocessor.json b/checkpoints/026000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/026000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/026000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/026000/pretrained_model/policy_preprocessor.json b/checkpoints/026000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/026000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/026000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/026000/pretrained_model/train_config.json b/checkpoints/026000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/026000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/026000/training_state/optimizer_param_groups.json b/checkpoints/026000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/026000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/026000/training_state/optimizer_state.safetensors b/checkpoints/026000/training_state/optimizer_state.safetensors deleted file mode 100644 index 79c4e668a8366e095dec9f210d28d15e2a671db9..0000000000000000000000000000000000000000 --- a/checkpoints/026000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:feb9311a2f55e34fed302dc5b61d0347654d2db290cd52d80342a30e287644a2 -size 412489972 diff --git a/checkpoints/026000/training_state/rng_state.safetensors b/checkpoints/026000/training_state/rng_state.safetensors deleted file mode 100644 index 258cf83a07cec1942a794bb34ae16bd661f022b9..0000000000000000000000000000000000000000 --- a/checkpoints/026000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:45ad066208ac65a08fbc923e0f8246eb4d022e06ed0e6ca7bcd33b50813e9902 -size 15708 diff --git a/checkpoints/026000/training_state/training_step.json b/checkpoints/026000/training_state/training_step.json deleted file mode 100644 index 60495b67a6a5c8ae0694d967a703afdf4bdc4066..0000000000000000000000000000000000000000 --- a/checkpoints/026000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 26000 -} \ No newline at end of file diff --git a/checkpoints/027000/pretrained_model/config.json b/checkpoints/027000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/027000/pretrained_model/model.safetensors b/checkpoints/027000/pretrained_model/model.safetensors deleted file mode 100644 index 8d93b3137717f22408ab808a5519131dc170962f..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:20ee5d3408669c856e48845715db573e8ad5ab2120a8888bfc5b256cfe1869fc -size 206372048 diff --git a/checkpoints/027000/pretrained_model/policy_postprocessor.json b/checkpoints/027000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/027000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/027000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/027000/pretrained_model/policy_preprocessor.json b/checkpoints/027000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/027000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/027000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/027000/pretrained_model/train_config.json b/checkpoints/027000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/027000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/027000/training_state/optimizer_param_groups.json b/checkpoints/027000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/027000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/027000/training_state/optimizer_state.safetensors b/checkpoints/027000/training_state/optimizer_state.safetensors deleted file mode 100644 index 54c8997e2140efb7d96d84b2b2e8bd771ca73655..0000000000000000000000000000000000000000 --- a/checkpoints/027000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:059777b39301c28f95eebbc39234aa8cd4786a9337a29780337a1ca4fe735097 -size 412489972 diff --git a/checkpoints/027000/training_state/rng_state.safetensors b/checkpoints/027000/training_state/rng_state.safetensors deleted file mode 100644 index bff3191e9df90f37c922e36ad68c4382c3fb2ce8..0000000000000000000000000000000000000000 --- a/checkpoints/027000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7e656190d99b02cb8d572669d37ca4193f6c3f7259fc1ebaa736fc5ce4c16fc6 -size 15708 diff --git a/checkpoints/027000/training_state/training_step.json b/checkpoints/027000/training_state/training_step.json deleted file mode 100644 index 0b7290592a849cf9c10e9fc2a06e13719ffd3a3c..0000000000000000000000000000000000000000 --- a/checkpoints/027000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 27000 -} \ No newline at end of file diff --git a/checkpoints/028000/pretrained_model/config.json b/checkpoints/028000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/028000/pretrained_model/model.safetensors b/checkpoints/028000/pretrained_model/model.safetensors deleted file mode 100644 index aa120414d7afefd1ab31dc74149be5b54c7cc664..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a3daaea6aad24b9adb2e8788fa612226e5f6c5ad375597266ec0a782f6365a4b -size 206372048 diff --git a/checkpoints/028000/pretrained_model/policy_postprocessor.json b/checkpoints/028000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/028000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/028000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/028000/pretrained_model/policy_preprocessor.json b/checkpoints/028000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/028000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/028000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/028000/pretrained_model/train_config.json b/checkpoints/028000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/028000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/028000/training_state/optimizer_param_groups.json b/checkpoints/028000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/028000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/028000/training_state/optimizer_state.safetensors b/checkpoints/028000/training_state/optimizer_state.safetensors deleted file mode 100644 index 8b42b6788e4a3b542b2eaf0e03213cd715885e16..0000000000000000000000000000000000000000 --- a/checkpoints/028000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:ba0bac2aa3c1a1fe7ab5d4b7c29457e82f4dee230c29222708e0d29d2278435e -size 412489972 diff --git a/checkpoints/028000/training_state/rng_state.safetensors b/checkpoints/028000/training_state/rng_state.safetensors deleted file mode 100644 index d98551f79ef01ca482b474e7e8ef7aeaded47225..0000000000000000000000000000000000000000 --- a/checkpoints/028000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2d24feda955b26f75050776d6f943adbb86b5756039bc0f5e94932aa9eec6061 -size 15708 diff --git a/checkpoints/028000/training_state/training_step.json b/checkpoints/028000/training_state/training_step.json deleted file mode 100644 index 86065a62b528a2c5fbe5615c4f34ea67322ce13e..0000000000000000000000000000000000000000 --- a/checkpoints/028000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 28000 -} \ No newline at end of file diff --git a/checkpoints/029000/pretrained_model/config.json b/checkpoints/029000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/029000/pretrained_model/model.safetensors b/checkpoints/029000/pretrained_model/model.safetensors deleted file mode 100644 index ce289ac784d8b86031d36aa75c1316eb80955d32..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a1195561ab50dc66dc9057273a3239dd7903b01b7b3ca8f4a270251fe450acfd -size 206372048 diff --git a/checkpoints/029000/pretrained_model/policy_postprocessor.json b/checkpoints/029000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/029000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/029000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/029000/pretrained_model/policy_preprocessor.json b/checkpoints/029000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/029000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/029000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/029000/pretrained_model/train_config.json b/checkpoints/029000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/029000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/029000/training_state/optimizer_param_groups.json b/checkpoints/029000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/029000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/029000/training_state/optimizer_state.safetensors b/checkpoints/029000/training_state/optimizer_state.safetensors deleted file mode 100644 index 8f74b5424e1aed9dde1cb99806793d1a7654c004..0000000000000000000000000000000000000000 --- a/checkpoints/029000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:e826d4de459713ebba27d84d50219cfbe8c6f8b2ca338a2342e5c88db0a6507a -size 412489972 diff --git a/checkpoints/029000/training_state/rng_state.safetensors b/checkpoints/029000/training_state/rng_state.safetensors deleted file mode 100644 index 441edcd3c925c3f33f76334543e763885bede98e..0000000000000000000000000000000000000000 --- a/checkpoints/029000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:aa531a70808c6a521cf0c503bfd7dd432daa3c082148f112d1681f2ef0bbf361 -size 15708 diff --git a/checkpoints/029000/training_state/training_step.json b/checkpoints/029000/training_state/training_step.json deleted file mode 100644 index b0ed16f946816dee5c395931f30b430702e4043a..0000000000000000000000000000000000000000 --- a/checkpoints/029000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 29000 -} \ No newline at end of file diff --git a/checkpoints/031000/pretrained_model/config.json b/checkpoints/031000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/031000/pretrained_model/model.safetensors b/checkpoints/031000/pretrained_model/model.safetensors deleted file mode 100644 index a1040e12502cf428065304c8260e6c68202532e0..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:fb6647d36ded01de8fd8816e7855a538752374e62066aba32e0dc88597211e48 -size 206372048 diff --git a/checkpoints/031000/pretrained_model/policy_postprocessor.json b/checkpoints/031000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/031000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/031000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/031000/pretrained_model/policy_preprocessor.json b/checkpoints/031000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/031000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/031000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/031000/pretrained_model/train_config.json b/checkpoints/031000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/031000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/031000/training_state/optimizer_param_groups.json b/checkpoints/031000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/031000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/031000/training_state/optimizer_state.safetensors b/checkpoints/031000/training_state/optimizer_state.safetensors deleted file mode 100644 index 2233770ced237be7bc0ec274a27e53056818d164..0000000000000000000000000000000000000000 --- a/checkpoints/031000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:67f584494e2df65a8f5d6799b5b1f62615333967e47756a6182b02c8ef5c8455 -size 412489972 diff --git a/checkpoints/031000/training_state/rng_state.safetensors b/checkpoints/031000/training_state/rng_state.safetensors deleted file mode 100644 index 144c3007a45f62cb0e29831cb7500caca79cc093..0000000000000000000000000000000000000000 --- a/checkpoints/031000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8f3f328e2fba2b3c9867e7272f0d7ca4b36fda1b04872eba5b67dfffa93f3281 -size 15708 diff --git a/checkpoints/031000/training_state/training_step.json b/checkpoints/031000/training_state/training_step.json deleted file mode 100644 index b336238f28b4df2bb873be706fc974549bc62f8c..0000000000000000000000000000000000000000 --- a/checkpoints/031000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 31000 -} \ No newline at end of file diff --git a/checkpoints/032000/pretrained_model/config.json b/checkpoints/032000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/032000/pretrained_model/model.safetensors b/checkpoints/032000/pretrained_model/model.safetensors deleted file mode 100644 index a6c4ea1a3d363a06ad2d53dcb71a6eb8f14f9f08..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:649c04ecdbf4c6cb8a3e80b48be870f5d6af58f6884272172bbd977dcd2c5d36 -size 206372048 diff --git a/checkpoints/032000/pretrained_model/policy_postprocessor.json b/checkpoints/032000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/032000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/032000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/032000/pretrained_model/policy_preprocessor.json b/checkpoints/032000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/032000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/032000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/032000/pretrained_model/train_config.json b/checkpoints/032000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/032000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/032000/training_state/optimizer_param_groups.json b/checkpoints/032000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/032000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/032000/training_state/optimizer_state.safetensors b/checkpoints/032000/training_state/optimizer_state.safetensors deleted file mode 100644 index eaa33060705b1069b91877fc8b7db412968bef12..0000000000000000000000000000000000000000 --- a/checkpoints/032000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f90291762de144df5d3e3dc6159f7ffd14ed64c30b2ff0f69790132e314efd00 -size 412489972 diff --git a/checkpoints/032000/training_state/rng_state.safetensors b/checkpoints/032000/training_state/rng_state.safetensors deleted file mode 100644 index 336d83a137f70a3b083fa887a26bc9f754285665..0000000000000000000000000000000000000000 --- a/checkpoints/032000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2226a8693e1d92af78e85df64b90461466cb61a645d88a99ffe5bd94bf72b72e -size 15708 diff --git a/checkpoints/032000/training_state/training_step.json b/checkpoints/032000/training_state/training_step.json deleted file mode 100644 index 360706c89342152bca05a3fe1d8c765a755213d2..0000000000000000000000000000000000000000 --- a/checkpoints/032000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 32000 -} \ No newline at end of file diff --git a/checkpoints/033000/pretrained_model/config.json b/checkpoints/033000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/033000/pretrained_model/model.safetensors b/checkpoints/033000/pretrained_model/model.safetensors deleted file mode 100644 index 4def2a3bf81b828210ab57705961c8796ac6de8e..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:9aa1cb7d614db03397486809b2626942d09a463321cd999f90f951959075a4a7 -size 206372048 diff --git a/checkpoints/033000/pretrained_model/policy_postprocessor.json b/checkpoints/033000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/033000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/033000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/033000/pretrained_model/policy_preprocessor.json b/checkpoints/033000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/033000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/033000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/033000/pretrained_model/train_config.json b/checkpoints/033000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/033000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/033000/training_state/optimizer_param_groups.json b/checkpoints/033000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/033000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/033000/training_state/optimizer_state.safetensors b/checkpoints/033000/training_state/optimizer_state.safetensors deleted file mode 100644 index 8bd6258519f9bb082d6a04442b8f6cdf239d5424..0000000000000000000000000000000000000000 --- a/checkpoints/033000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f1e4654bdda5ec622403353651571093b53f24b255bfb6fcc1ee8fd3c7815d2a -size 412489972 diff --git a/checkpoints/033000/training_state/rng_state.safetensors b/checkpoints/033000/training_state/rng_state.safetensors deleted file mode 100644 index 73fafa03e030ada3fe7fe252f81ff5e23120c30a..0000000000000000000000000000000000000000 --- a/checkpoints/033000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:e5a7997ae3f510ef315b2c66d7f2a818a07d3a643d045e5aa5bf5daa8fea9d84 -size 15708 diff --git a/checkpoints/033000/training_state/training_step.json b/checkpoints/033000/training_state/training_step.json deleted file mode 100644 index f1a8ee148932c4dcd38c2709874939717bade74e..0000000000000000000000000000000000000000 --- a/checkpoints/033000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 33000 -} \ No newline at end of file diff --git a/checkpoints/034000/pretrained_model/config.json b/checkpoints/034000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/034000/pretrained_model/model.safetensors b/checkpoints/034000/pretrained_model/model.safetensors deleted file mode 100644 index f21e4f8663675407b61573052439856e4aa62678..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a20b0fada4083ea037ee4018f99f1ff8f1606afa46c1166c024377d226693d12 -size 206372048 diff --git a/checkpoints/034000/pretrained_model/policy_postprocessor.json b/checkpoints/034000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/034000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/034000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/034000/pretrained_model/policy_preprocessor.json b/checkpoints/034000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/034000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/034000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/034000/pretrained_model/train_config.json b/checkpoints/034000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/034000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/034000/training_state/optimizer_param_groups.json b/checkpoints/034000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/034000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/034000/training_state/optimizer_state.safetensors b/checkpoints/034000/training_state/optimizer_state.safetensors deleted file mode 100644 index 2f045fc9f2237c12c276c6a6b4e86035fc1cd398..0000000000000000000000000000000000000000 --- a/checkpoints/034000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:d34f3f37de359bfa798fcfdaf6f8c384c37445045efca5b9f23ea7b5ed222990 -size 412489972 diff --git a/checkpoints/034000/training_state/rng_state.safetensors b/checkpoints/034000/training_state/rng_state.safetensors deleted file mode 100644 index e496b210fa71a228912e26a37afa0e2126ca450a..0000000000000000000000000000000000000000 --- a/checkpoints/034000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:c938e209d85087ab07d56f08d3c69f011a3b928ddc91c372d6bdbb7cde7a8f4c -size 15708 diff --git a/checkpoints/034000/training_state/training_step.json b/checkpoints/034000/training_state/training_step.json deleted file mode 100644 index 740e2f1334f5726ddfb1efb9eb0cf8def5a503cd..0000000000000000000000000000000000000000 --- a/checkpoints/034000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 34000 -} \ No newline at end of file diff --git a/checkpoints/036000/pretrained_model/config.json b/checkpoints/036000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/036000/pretrained_model/model.safetensors b/checkpoints/036000/pretrained_model/model.safetensors deleted file mode 100644 index 7fdc794ff29f4498cb359ee6599d70e21d609d18..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:602b120c2f62a05ffb89f6df1471d845190b47321745ed4f87f48d07399c8a25 -size 206372048 diff --git a/checkpoints/036000/pretrained_model/policy_postprocessor.json b/checkpoints/036000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/036000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/036000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/036000/pretrained_model/policy_preprocessor.json b/checkpoints/036000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/036000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/036000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/036000/pretrained_model/train_config.json b/checkpoints/036000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/036000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/036000/training_state/optimizer_param_groups.json b/checkpoints/036000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/036000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/036000/training_state/optimizer_state.safetensors b/checkpoints/036000/training_state/optimizer_state.safetensors deleted file mode 100644 index e1cc10012415f173ee7c137d1b500baf419d73b5..0000000000000000000000000000000000000000 --- a/checkpoints/036000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:d7f16294cf0618114615e5991599ca943513db10fb5126905d69e4068bbb35e8 -size 412489972 diff --git a/checkpoints/036000/training_state/rng_state.safetensors b/checkpoints/036000/training_state/rng_state.safetensors deleted file mode 100644 index 8c73439daff2159f94182af90a9e678539999194..0000000000000000000000000000000000000000 --- a/checkpoints/036000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a2ca9a79a6f42c6df6866a03b4c3240275465c37bc1f1e70748bbb061e70b0a1 -size 15708 diff --git a/checkpoints/036000/training_state/training_step.json b/checkpoints/036000/training_state/training_step.json deleted file mode 100644 index ec74060efe48f3deb933d42c46542fa562986f50..0000000000000000000000000000000000000000 --- a/checkpoints/036000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 36000 -} \ No newline at end of file diff --git a/checkpoints/037000/pretrained_model/config.json b/checkpoints/037000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/037000/pretrained_model/model.safetensors b/checkpoints/037000/pretrained_model/model.safetensors deleted file mode 100644 index b0ae1f87ee30c93f56703d513790f3a2b720a26d..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:627d1e2608da1a4aa3564a1b798e37e6fb7cc3ff8d091eb0bfe25dd6b8b1ba1a -size 206372048 diff --git a/checkpoints/037000/pretrained_model/policy_postprocessor.json b/checkpoints/037000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/037000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/037000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/037000/pretrained_model/policy_preprocessor.json b/checkpoints/037000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/037000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/037000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/037000/pretrained_model/train_config.json b/checkpoints/037000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/037000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/037000/training_state/optimizer_param_groups.json b/checkpoints/037000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/037000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/037000/training_state/optimizer_state.safetensors b/checkpoints/037000/training_state/optimizer_state.safetensors deleted file mode 100644 index 3ff65ebb1892a987d31315f25140547f96e4462b..0000000000000000000000000000000000000000 --- a/checkpoints/037000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:27ac4a074ba0756842042218bc79221a706c6121779514f28b702c438ba1e866 -size 412489972 diff --git a/checkpoints/037000/training_state/rng_state.safetensors b/checkpoints/037000/training_state/rng_state.safetensors deleted file mode 100644 index 67051ea0389120404a3a258aa53a16a6016f86d4..0000000000000000000000000000000000000000 --- a/checkpoints/037000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2d12fe40f519a8d14606f770e13883524a2868a56a1c1f355b04a5e652af634d -size 15708 diff --git a/checkpoints/037000/training_state/training_step.json b/checkpoints/037000/training_state/training_step.json deleted file mode 100644 index 59adb96582f1246b398f54d3e8cca0cc4f97e7c2..0000000000000000000000000000000000000000 --- a/checkpoints/037000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 37000 -} \ No newline at end of file diff --git a/checkpoints/038000/pretrained_model/config.json b/checkpoints/038000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/038000/pretrained_model/model.safetensors b/checkpoints/038000/pretrained_model/model.safetensors deleted file mode 100644 index 6cd1b0a2214bf4493c624bab368ba038ff586ec3..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:9afd92a9a051638d5615e6785f0d2d897904a7a6a4237bfe4ea1ce7b284db684 -size 206372048 diff --git a/checkpoints/038000/pretrained_model/policy_postprocessor.json b/checkpoints/038000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/038000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/038000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/038000/pretrained_model/policy_preprocessor.json b/checkpoints/038000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/038000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/038000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/038000/pretrained_model/train_config.json b/checkpoints/038000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/038000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/038000/training_state/optimizer_param_groups.json b/checkpoints/038000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/038000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/038000/training_state/optimizer_state.safetensors b/checkpoints/038000/training_state/optimizer_state.safetensors deleted file mode 100644 index 6a46948ab897ea5434845dba09ece419fa1cf6f2..0000000000000000000000000000000000000000 --- a/checkpoints/038000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:dd30f863a276907f5e6bdb8bb07a63011efac57713f265d8cc0c245db2ebee14 -size 412489972 diff --git a/checkpoints/038000/training_state/rng_state.safetensors b/checkpoints/038000/training_state/rng_state.safetensors deleted file mode 100644 index 4844a35d1b93d2e9c721d7c521b2246a38bc8d4a..0000000000000000000000000000000000000000 --- a/checkpoints/038000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:fd0fcfc86029ef30cf8c149528811196b6b5ba960ed6427f334e95c5011fb0be -size 15708 diff --git a/checkpoints/038000/training_state/training_step.json b/checkpoints/038000/training_state/training_step.json deleted file mode 100644 index d3d23fa7ab917d4467e871a903d8b35e962ef1c8..0000000000000000000000000000000000000000 --- a/checkpoints/038000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 38000 -} \ No newline at end of file diff --git a/checkpoints/039000/pretrained_model/config.json b/checkpoints/039000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/039000/pretrained_model/model.safetensors b/checkpoints/039000/pretrained_model/model.safetensors deleted file mode 100644 index 14bd75eb421605c6f2204bae9f5fabe0359ddf55..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8b52baaf21395d35c3ae0ce522820de697f04248afd8d6ccd335afa5d60984d2 -size 206372048 diff --git a/checkpoints/039000/pretrained_model/policy_postprocessor.json b/checkpoints/039000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/039000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/039000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/039000/pretrained_model/policy_preprocessor.json b/checkpoints/039000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/039000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/039000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/039000/pretrained_model/train_config.json b/checkpoints/039000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/039000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/039000/training_state/optimizer_param_groups.json b/checkpoints/039000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/039000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/039000/training_state/optimizer_state.safetensors b/checkpoints/039000/training_state/optimizer_state.safetensors deleted file mode 100644 index 3bde74c54bc0e93f7117c28d6489e065ae88261a..0000000000000000000000000000000000000000 --- a/checkpoints/039000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:bb9a5ce00ee83dbf0b3d147954da8cd4fd6a4b98aeb2ae454b351987f3bae832 -size 412489972 diff --git a/checkpoints/039000/training_state/rng_state.safetensors b/checkpoints/039000/training_state/rng_state.safetensors deleted file mode 100644 index b201077b061bc28a25ad4fea9e9c212ab9541312..0000000000000000000000000000000000000000 --- a/checkpoints/039000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:410eda79efed7f30779d8b1bcc92f58074f3092df7c18f2393db1e30b3cb5642 -size 15708 diff --git a/checkpoints/039000/training_state/training_step.json b/checkpoints/039000/training_state/training_step.json deleted file mode 100644 index 8fbb17ce1a59c5d1e6fef40fc2e96147818d5645..0000000000000000000000000000000000000000 --- a/checkpoints/039000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 39000 -} \ No newline at end of file diff --git a/checkpoints/041000/pretrained_model/config.json b/checkpoints/041000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/041000/pretrained_model/model.safetensors b/checkpoints/041000/pretrained_model/model.safetensors deleted file mode 100644 index e86336b6ffea914ab9c5ce2f2dddb1b47d21f958..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2befe3de3b6be68f4ef1e24a7738e6a95e8ca0e5e9a3d42afb3fef9ac587b88f -size 206372048 diff --git a/checkpoints/041000/pretrained_model/policy_postprocessor.json b/checkpoints/041000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/041000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/041000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/041000/pretrained_model/policy_preprocessor.json b/checkpoints/041000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/041000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/041000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/041000/pretrained_model/train_config.json b/checkpoints/041000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/041000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/041000/training_state/optimizer_param_groups.json b/checkpoints/041000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/041000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/041000/training_state/optimizer_state.safetensors b/checkpoints/041000/training_state/optimizer_state.safetensors deleted file mode 100644 index 8c72eaab03f6081c18a3cd073f7ebaee16be99bf..0000000000000000000000000000000000000000 --- a/checkpoints/041000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:1e0ba31a66541f9a5af78f4078b82a4ade28cabedf59da45c2267f07ad7402e0 -size 412489972 diff --git a/checkpoints/041000/training_state/rng_state.safetensors b/checkpoints/041000/training_state/rng_state.safetensors deleted file mode 100644 index d9bdf5a1bd770a20726e09e9ca9884d6d6244bf3..0000000000000000000000000000000000000000 --- a/checkpoints/041000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:716e2202955bfce090e38681f1976b6881fc4199f8073829304c03888fd06056 -size 15708 diff --git a/checkpoints/041000/training_state/training_step.json b/checkpoints/041000/training_state/training_step.json deleted file mode 100644 index 71a28153cbe44488b9e823e633ccf889e28b7a96..0000000000000000000000000000000000000000 --- a/checkpoints/041000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 41000 -} \ No newline at end of file diff --git a/checkpoints/042000/pretrained_model/config.json b/checkpoints/042000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/042000/pretrained_model/model.safetensors b/checkpoints/042000/pretrained_model/model.safetensors deleted file mode 100644 index c13d7f33e9d878d6a694ade9527f3243af3f4440..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:da2c024ac05764a5135f479c61a0e964f20aba82e3269a92a8ba51ec36ebf38e -size 206372048 diff --git a/checkpoints/042000/pretrained_model/policy_postprocessor.json b/checkpoints/042000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/042000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/042000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/042000/pretrained_model/policy_preprocessor.json b/checkpoints/042000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/042000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/042000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/042000/pretrained_model/train_config.json b/checkpoints/042000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/042000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/042000/training_state/optimizer_param_groups.json b/checkpoints/042000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/042000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/042000/training_state/optimizer_state.safetensors b/checkpoints/042000/training_state/optimizer_state.safetensors deleted file mode 100644 index d7fba2934f60fe82283a566bba3d083c7782ec0b..0000000000000000000000000000000000000000 --- a/checkpoints/042000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:99c9a87d7326b508a5d338ddd0d02fdb10274a44d731eb00426b084cf7a11d5b -size 412489972 diff --git a/checkpoints/042000/training_state/rng_state.safetensors b/checkpoints/042000/training_state/rng_state.safetensors deleted file mode 100644 index cea1f9ceb0884e4bf8bf9a64dafe9774c9c6f9d8..0000000000000000000000000000000000000000 --- a/checkpoints/042000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f48821105065e26553c3d723fec1106d4ad6456ff161a9c1e16eb9ec8c96a4ae -size 15708 diff --git a/checkpoints/042000/training_state/training_step.json b/checkpoints/042000/training_state/training_step.json deleted file mode 100644 index 68f7e50d12ca49bcf4a2afe411c5215ffaf76dfa..0000000000000000000000000000000000000000 --- a/checkpoints/042000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 42000 -} \ No newline at end of file diff --git a/checkpoints/043000/pretrained_model/config.json b/checkpoints/043000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/043000/pretrained_model/model.safetensors b/checkpoints/043000/pretrained_model/model.safetensors deleted file mode 100644 index 6a9c210c8f4102e9c63a1846a2148b0ab0395542..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8c7ea2d55b3dca10d5d08bdcb7b21e162dc2e4d0649a6e258dd3095d5e7a712d -size 206372048 diff --git a/checkpoints/043000/pretrained_model/policy_postprocessor.json b/checkpoints/043000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/043000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/043000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/043000/pretrained_model/policy_preprocessor.json b/checkpoints/043000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/043000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/043000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/043000/pretrained_model/train_config.json b/checkpoints/043000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/043000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/043000/training_state/optimizer_param_groups.json b/checkpoints/043000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/043000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/043000/training_state/optimizer_state.safetensors b/checkpoints/043000/training_state/optimizer_state.safetensors deleted file mode 100644 index 98b8774b64748c1dfab0613b3f9af8b00118c25b..0000000000000000000000000000000000000000 --- a/checkpoints/043000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:9a21b13727aa20898314aa979bc5af97f20a2bb3fa41dae3a453cadb6e70de42 -size 412489972 diff --git a/checkpoints/043000/training_state/rng_state.safetensors b/checkpoints/043000/training_state/rng_state.safetensors deleted file mode 100644 index 0cc3a21e8ef411927f6f01901fe78f2548954f1f..0000000000000000000000000000000000000000 --- a/checkpoints/043000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:f629e5817ce4ef16b3b8e64cd6c3f34f717cc55f36a20d2bdd4768e4a4cce3fc -size 15708 diff --git a/checkpoints/043000/training_state/training_step.json b/checkpoints/043000/training_state/training_step.json deleted file mode 100644 index 2c813616877b02381c37abe6f4a79758d6ebb915..0000000000000000000000000000000000000000 --- a/checkpoints/043000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 43000 -} \ No newline at end of file diff --git a/checkpoints/044000/pretrained_model/config.json b/checkpoints/044000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/044000/pretrained_model/model.safetensors b/checkpoints/044000/pretrained_model/model.safetensors deleted file mode 100644 index 2b774bfd8d53ef6387436d11da5420f1a8b39b2f..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2f20389f7dbcbffce6a87d7f45a4f5e789c5e4b2f2021e94b70cc652a7adcd73 -size 206372048 diff --git a/checkpoints/044000/pretrained_model/policy_postprocessor.json b/checkpoints/044000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/044000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/044000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/044000/pretrained_model/policy_preprocessor.json b/checkpoints/044000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/044000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/044000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/044000/pretrained_model/train_config.json b/checkpoints/044000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/044000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/044000/training_state/optimizer_param_groups.json b/checkpoints/044000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/044000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/044000/training_state/optimizer_state.safetensors b/checkpoints/044000/training_state/optimizer_state.safetensors deleted file mode 100644 index 7797e145f11466d662ece6960eb061bb2bc8f651..0000000000000000000000000000000000000000 --- a/checkpoints/044000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:d964984b8da061af4b08ac2fb0e46a499e734fb889647c837e41f8374619338c -size 412489972 diff --git a/checkpoints/044000/training_state/rng_state.safetensors b/checkpoints/044000/training_state/rng_state.safetensors deleted file mode 100644 index 85550143dfd1f980706d1a53737eacc486998783..0000000000000000000000000000000000000000 --- a/checkpoints/044000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:5908dee9ed3a5ecf1812b9ec0df8aa479d6744e8f4c43fd85b7f15ae23e6ebd5 -size 15708 diff --git a/checkpoints/044000/training_state/training_step.json b/checkpoints/044000/training_state/training_step.json deleted file mode 100644 index 6a7afd8c0e19d8169d11b8b87964422d193ca923..0000000000000000000000000000000000000000 --- a/checkpoints/044000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 44000 -} \ No newline at end of file diff --git a/checkpoints/046000/pretrained_model/config.json b/checkpoints/046000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/046000/pretrained_model/model.safetensors b/checkpoints/046000/pretrained_model/model.safetensors deleted file mode 100644 index ce2d01acc40eaecb20715d8017fb97f037c5fcb3..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:56eabf6b5071aea2c3a5ffc6f707d85f493c5f536064b0e10f8b784097f39e3f -size 206372048 diff --git a/checkpoints/046000/pretrained_model/policy_postprocessor.json b/checkpoints/046000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/046000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/046000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/046000/pretrained_model/policy_preprocessor.json b/checkpoints/046000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/046000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/046000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/046000/pretrained_model/train_config.json b/checkpoints/046000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/046000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/046000/training_state/optimizer_param_groups.json b/checkpoints/046000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/046000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/046000/training_state/optimizer_state.safetensors b/checkpoints/046000/training_state/optimizer_state.safetensors deleted file mode 100644 index 1e59d537225e61b670985dc4d8aac1e9f1dd6823..0000000000000000000000000000000000000000 --- a/checkpoints/046000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:2e586a71688090a18929242ef9f9b5538116ef4e6f1b78ec119bb0b25c4a5c84 -size 412489972 diff --git a/checkpoints/046000/training_state/rng_state.safetensors b/checkpoints/046000/training_state/rng_state.safetensors deleted file mode 100644 index f2418fd70cdb443fe64380cd91f1c04dd6082b36..0000000000000000000000000000000000000000 --- a/checkpoints/046000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:1c713e7e1f86bca214156e7220c210a22451e95820801a7704c95f06746bdf6c -size 15708 diff --git a/checkpoints/046000/training_state/training_step.json b/checkpoints/046000/training_state/training_step.json deleted file mode 100644 index a43ac865a273a5865dc7d1f28d4f9a078f67abe0..0000000000000000000000000000000000000000 --- a/checkpoints/046000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 46000 -} \ No newline at end of file diff --git a/checkpoints/047000/pretrained_model/config.json b/checkpoints/047000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/047000/pretrained_model/model.safetensors b/checkpoints/047000/pretrained_model/model.safetensors deleted file mode 100644 index 647f2615c2ad20999384c6354f0631e292da777b..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:6275c94fc1879fbb6081848b41ebfcb80ad15f68cd83a41c70bead32d638a881 -size 206372048 diff --git a/checkpoints/047000/pretrained_model/policy_postprocessor.json b/checkpoints/047000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/047000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/047000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/047000/pretrained_model/policy_preprocessor.json b/checkpoints/047000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/047000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/047000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/047000/pretrained_model/train_config.json b/checkpoints/047000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/047000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/047000/training_state/optimizer_param_groups.json b/checkpoints/047000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/047000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/047000/training_state/optimizer_state.safetensors b/checkpoints/047000/training_state/optimizer_state.safetensors deleted file mode 100644 index f8f04f62436aa982741f35d67efe04ccd1bbca00..0000000000000000000000000000000000000000 --- a/checkpoints/047000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:be48cf26dac796f30b9a4158a2304d154a9f067fb8a54e846e2bbefa91071678 -size 412489972 diff --git a/checkpoints/047000/training_state/rng_state.safetensors b/checkpoints/047000/training_state/rng_state.safetensors deleted file mode 100644 index ac76ee9986916a91d5e4efe1cc6b78328c1fc08e..0000000000000000000000000000000000000000 --- a/checkpoints/047000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:8d5579de093188efb084be0a0a88eebe9dfdf66da5c5895cdd628946c65614bc -size 15708 diff --git a/checkpoints/047000/training_state/training_step.json b/checkpoints/047000/training_state/training_step.json deleted file mode 100644 index 85db166430962b5b70bc2494f7a747a4565772ed..0000000000000000000000000000000000000000 --- a/checkpoints/047000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 47000 -} \ No newline at end of file diff --git a/checkpoints/048000/pretrained_model/config.json b/checkpoints/048000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/048000/pretrained_model/model.safetensors b/checkpoints/048000/pretrained_model/model.safetensors deleted file mode 100644 index 0f8a779766f43cc7e63ef3e449c733709bbe84d9..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:00bbd7f4fe5e4f9b48b5f8a538649b221beae9b54c845f0ef8718161a38ca2f7 -size 206372048 diff --git a/checkpoints/048000/pretrained_model/policy_postprocessor.json b/checkpoints/048000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/048000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/048000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/048000/pretrained_model/policy_preprocessor.json b/checkpoints/048000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/048000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/048000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/048000/pretrained_model/train_config.json b/checkpoints/048000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/048000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/048000/training_state/optimizer_param_groups.json b/checkpoints/048000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/048000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/048000/training_state/optimizer_state.safetensors b/checkpoints/048000/training_state/optimizer_state.safetensors deleted file mode 100644 index 9ee97de60ff9a4a7e76eb64eb5de5791991bf44d..0000000000000000000000000000000000000000 --- a/checkpoints/048000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7fd7c24bbfcbb1f0a18009db82bf3ff59c4cc9dc665d84ccbabcbfb760f41345 -size 412489972 diff --git a/checkpoints/048000/training_state/rng_state.safetensors b/checkpoints/048000/training_state/rng_state.safetensors deleted file mode 100644 index faa4b1d848ceb16d42e7a43b69546548020b6781..0000000000000000000000000000000000000000 --- a/checkpoints/048000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:b23aae3a247525debe60ee5263b2b6d3e3caaabd577ec846255f997ee1aea9d4 -size 15708 diff --git a/checkpoints/048000/training_state/training_step.json b/checkpoints/048000/training_state/training_step.json deleted file mode 100644 index 8a90a8c25646098fb894b4970773a854c286dd33..0000000000000000000000000000000000000000 --- a/checkpoints/048000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 48000 -} \ No newline at end of file diff --git a/checkpoints/049000/pretrained_model/config.json b/checkpoints/049000/pretrained_model/config.json deleted file mode 100644 index 4c70568529c9cccb7ad96000c378503b8e04e4fb..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/config.json +++ /dev/null @@ -1,70 +0,0 @@ -{ - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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/checkpoints/049000/pretrained_model/model.safetensors b/checkpoints/049000/pretrained_model/model.safetensors deleted file mode 100644 index 89ead54792404bf3ac1d7cabc6a2fc558060966a..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/model.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:40ead7bdcdf00ad8cca749efd8bcc4acf3b51f418bdd63eb9b82dccb8358a5a6 -size 206372048 diff --git a/checkpoints/049000/pretrained_model/policy_postprocessor.json b/checkpoints/049000/pretrained_model/policy_postprocessor.json deleted file mode 100644 index 9ccb342474cecbf2bdbb034d27169a351ad96d2a..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/policy_postprocessor.json +++ /dev/null @@ -1,32 +0,0 @@ -{ - "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/checkpoints/049000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors b/checkpoints/049000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/049000/pretrained_model/policy_preprocessor.json b/checkpoints/049000/pretrained_model/policy_preprocessor.json deleted file mode 100644 index 03d05b4b946715ecac97a9e38bd20b62f769fd8b..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/policy_preprocessor.json +++ /dev/null @@ -1,64 +0,0 @@ -{ - "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": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "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/checkpoints/049000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors b/checkpoints/049000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors deleted file mode 100644 index 5d7832a8fa96a75f79496a200dcb5059a9e05aaa..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:21de742308f7b192bf33f8be618ce291683f0b6d78c013f09b6f0398edda81fc -size 7584 diff --git a/checkpoints/049000/pretrained_model/train_config.json b/checkpoints/049000/pretrained_model/train_config.json deleted file mode 100644 index 8d2e747c9e301af4cec7cf54965d1964ec4d2807..0000000000000000000000000000000000000000 --- a/checkpoints/049000/pretrained_model/train_config.json +++ /dev/null @@ -1,193 +0,0 @@ -{ - "dataset": { - "repo_id": "abdul004/so101_ball_in_cup_v4", - "root": null, - "episodes": null, - "image_transforms": { - "enable": false, - "max_num_transforms": 3, - "random_order": false, - "tfs": { - "brightness": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "brightness": [ - 0.8, - 1.2 - ] - } - }, - "contrast": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "contrast": [ - 0.8, - 1.2 - ] - } - }, - "saturation": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "saturation": [ - 0.5, - 1.5 - ] - } - }, - "hue": { - "weight": 1.0, - "type": "ColorJitter", - "kwargs": { - "hue": [ - -0.05, - 0.05 - ] - } - }, - "sharpness": { - "weight": 1.0, - "type": "SharpnessJitter", - "kwargs": { - "sharpness": [ - 0.5, - 1.5 - ] - } - }, - "affine": { - "weight": 1.0, - "type": "RandomAffine", - "kwargs": { - "degrees": [ - -5.0, - 5.0 - ], - "translate": [ - 0.05, - 0.05 - ] - } - } - } - }, - "revision": null, - "use_imagenet_stats": true, - "video_backend": "torchcodec", - "streaming": false - }, - "env": null, - "policy": { - "type": "act", - "n_obs_steps": 1, - "input_features": { - "observation.state": { - "type": "STATE", - "shape": [ - 6 - ] - }, - "observation.images.wrist": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - }, - "observation.images.overhead": { - "type": "VISUAL", - "shape": [ - 3, - 480, - 640 - ] - } - }, - "output_features": { - "action": { - "type": "ACTION", - "shape": [ - 6 - ] - } - }, - "device": "cuda", - "use_amp": false, - "push_to_hub": true, - "repo_id": "abdul004/so101_act_policy_v4", - "private": null, - "tags": null, - "license": null, - "pretrained_path": "resume_ckpt/checkpoints/002000/pretrained_model", - "chunk_size": 20, - "n_action_steps": 20, - "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": 0, - "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 - }, - "output_dir": "outputs/train/so101_v4_act", - "job_name": "act", - "resume": false, - "seed": 1000, - "num_workers": 4, - "batch_size": 32, - "steps": 50000, - "eval_freq": -1, - "log_freq": 100, - "save_checkpoint": true, - "save_freq": 1000, - "use_policy_training_preset": true, - "optimizer": { - "type": "adamw", - "lr": 1e-05, - "weight_decay": 0.0001, - "grad_clip_norm": 10.0, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08 - }, - "scheduler": null, - "eval": { - "n_episodes": 50, - "batch_size": 50, - "use_async_envs": false - }, - "wandb": { - "enable": false, - "disable_artifact": false, - "project": "lerobot", - "entity": null, - "notes": null, - "run_id": null, - "mode": null - }, - "checkpoint_path": null, - "rename_map": {} -} \ No newline at end of file diff --git a/checkpoints/049000/training_state/optimizer_param_groups.json b/checkpoints/049000/training_state/optimizer_param_groups.json deleted file mode 100644 index 33506df097b87c64b2aa15765ab89706780c0dc0..0000000000000000000000000000000000000000 --- a/checkpoints/049000/training_state/optimizer_param_groups.json +++ /dev/null @@ -1,191 +0,0 @@ -[ - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 0, - 1, - 2, - 3, - 4, - 5, - 6, - 7, - 8, - 9, - 10, - 11, - 12, - 13, - 14, - 15, - 16, - 17, - 18, - 19, - 20, - 21, - 22, - 23, - 24, - 25, - 26, - 27, - 28, - 29, - 30, - 31, - 32, - 33, - 34, - 35, - 36, - 37, - 38, - 39, - 40, - 41, - 42, - 43, - 44, - 45, - 46, - 47, - 48, - 49, - 50, - 51, - 52, - 53, - 54, - 55, - 56, - 57, - 58, - 59, - 60, - 61, - 62, - 63, - 64, - 65, - 66, - 67, - 68, - 69, - 70, - 71, - 72, - 73, - 74, - 75, - 76, - 77, - 78, - 79, - 80, - 81, - 82, - 83, - 84, - 85, - 86, - 87, - 88, - 89, - 90, - 91, - 92, - 93, - 94, - 95, - 96, - 97, - 98, - 99, - 100, - 101, - 102, - 103, - 104, - 105, - 106, - 107, - 108, - 109, - 110, - 111, - 112, - 113, - 114, - 115, - 116, - 117, - 118, - 119, - 120, - 121, - 122, - 123, - 124, - 125, - 126, - 127, - 128, - 129, - 130, - 131, - 132 - ] - }, - { - "lr": 1e-05, - "betas": [ - 0.9, - 0.999 - ], - "eps": 1e-08, - "weight_decay": 0.0001, - "amsgrad": false, - "maximize": false, - "foreach": null, - "capturable": false, - "differentiable": false, - "fused": null, - "decoupled_weight_decay": true, - "params": [ - 133, - 134, - 135, - 136, - 137, - 138, - 139, - 140, - 141, - 142, - 143, - 144, - 145, - 146, - 147, - 148, - 149, - 150, - 151, - 152 - ] - } -] \ No newline at end of file diff --git a/checkpoints/049000/training_state/optimizer_state.safetensors b/checkpoints/049000/training_state/optimizer_state.safetensors deleted file mode 100644 index 0ef405080b0686c9315e772ad9e2805051443330..0000000000000000000000000000000000000000 --- a/checkpoints/049000/training_state/optimizer_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:1d22ad5a9502f61c33b1e9656a6a14920138b43df0420ac05e924795da8e9da1 -size 412489972 diff --git a/checkpoints/049000/training_state/rng_state.safetensors b/checkpoints/049000/training_state/rng_state.safetensors deleted file mode 100644 index c47e505aaf8d5ad889da776bd98c0294a300e655..0000000000000000000000000000000000000000 --- a/checkpoints/049000/training_state/rng_state.safetensors +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:7e3d67dde998f87c32429ec272f47942e7d52e5b48ff1dec112dcde0342e7b93 -size 15708 diff --git a/checkpoints/049000/training_state/training_step.json b/checkpoints/049000/training_state/training_step.json deleted file mode 100644 index d37e06b6733ad5a77ddd6be83331fbafbcaa143d..0000000000000000000000000000000000000000 --- a/checkpoints/049000/training_state/training_step.json +++ /dev/null @@ -1,3 +0,0 @@ -{ - "step": 49000 -} \ No newline at end of file