Remove superseded checkpoint 010000

checkpoints/010000/pretrained_model/config.json

@@ -1,106 +0,0 @@
-{
-    "type": "multi_task_dit",
-    "n_obs_steps": 2,
-    "input_features": {
-        "observation.state": {
-            "type": "STATE",
-            "shape": [
-                7
-            ]
-        },
-        "observation.images.cam1": {
-            "type": "VISUAL",
-            "shape": [
-                3,
-                720,
-                1280
-            ]
-        },
-        "observation.images.cam2": {
-            "type": "VISUAL",
-            "shape": [
-                3,
-                720,
-                1280
-            ]
-        }
-    },
-    "output_features": {
-        "action": {
-            "type": "ACTION",
-            "shape": [
-                7
-            ]
-        }
-    },
-    "device": "cuda",
-    "use_amp": false,
-    "use_peft": false,
-    "push_to_hub": false,
-    "repo_id": null,
-    "private": null,
-    "tags": null,
-    "license": null,
-    "pretrained_path": null,
-    "pretrained_revision": null,
-    "horizon": 32,
-    "n_action_steps": 24,
-    "objective": "flow_matching",
-    "noise_scheduler_type": "DDPM",
-    "num_train_timesteps": 100,
-    "beta_schedule": "squaredcos_cap_v2",
-    "beta_start": 0.0001,
-    "beta_end": 0.02,
-    "prediction_type": "epsilon",
-    "clip_sample": true,
-    "clip_sample_range": 1.0,
-    "num_inference_steps": null,
-    "sigma_min": 0.0,
-    "num_integration_steps": 100,
-    "integration_method": "euler",
-    "timestep_sampling_strategy": "beta",
-    "timestep_sampling_s": 0.999,
-    "timestep_sampling_alpha": 1.5,
-    "timestep_sampling_beta": 1.0,
-    "hidden_dim": 512,
-    "num_layers": 6,
-    "num_heads": 8,
-    "dropout": 0.1,
-    "use_positional_encoding": false,
-    "timestep_embed_dim": 256,
-    "use_rope": true,
-    "rope_base": 10000.0,
-    "vision_encoder_name": "openai/clip-vit-base-patch16",
-    "use_separate_rgb_encoder_per_camera": false,
-    "vision_encoder_lr_multiplier": 0.1,
-    "image_resize_shape": [
-        224,
-        224
-    ],
-    "image_crop_shape": [
-        224,
-        224
-    ],
-    "image_crop_is_random": true,
-    "text_encoder_name": "openai/clip-vit-base-patch16",
-    "tokenizer_max_length": 77,
-    "tokenizer_padding": "max_length",
-    "tokenizer_padding_side": "right",
-    "tokenizer_truncation": true,
-    "normalization_mapping": {
-        "VISUAL": "MEAN_STD",
-        "STATE": "MIN_MAX",
-        "ACTION": "MIN_MAX"
-    },
-    "optimizer_lr": 2e-05,
-    "optimizer_betas": [
-        0.95,
-        0.999
-    ],
-    "optimizer_eps": 1e-08,
-    "optimizer_weight_decay": 0.0,
-    "scheduler_name": "cosine",
-    "scheduler_warmup_steps": 0,
-    "do_mask_loss_for_padding": false,
-    "drop_n_last_frames": 7
-}

checkpoints/010000/pretrained_model/model.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:898da5d3ffa06e243548921ccb7e9a2e3bfdc7fc95d0833f4dbc1bd88228a29f
-size 995640924

checkpoints/010000/pretrained_model/policy_postprocessor.json

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

checkpoints/010000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:ebcf8788ac2c38ccae8bc139e57a83da88fa38f6ab01a8fcc40cdbccdeb5286b
-size 7608

checkpoints/010000/pretrained_model/policy_preprocessor.json

@@ -1,75 +0,0 @@
-{
-  "name": "policy_preprocessor",
-  "steps": [
-    {
-      "registry_name": "rename_observations_processor",
-      "config": {
-        "rename_map": {}
-      }
-    },
-    {
-      "registry_name": "to_batch_processor",
-      "config": {}
-    },
-    {
-      "registry_name": "tokenizer_processor",
-      "config": {
-        "max_length": 77,
-        "task_key": "task",
-        "padding_side": "right",
-        "padding": "max_length",
-        "truncation": true,
-        "tokenizer_name": "openai/clip-vit-base-patch16"
-      }
-    },
-    {
-      "registry_name": "device_processor",
-      "config": {
-        "device": "cuda",
-        "float_dtype": null
-      }
-    },
-    {
-      "registry_name": "normalizer_processor",
-      "config": {
-        "eps": 1e-08,
-        "features": {
-          "observation.state": {
-            "type": "STATE",
-            "shape": [
-              7
-            ]
-          },
-          "observation.images.cam1": {
-            "type": "VISUAL",
-            "shape": [
-              3,
-              720,
-              1280
-            ]
-          },
-          "observation.images.cam2": {
-            "type": "VISUAL",
-            "shape": [
-              3,
-              720,
-              1280
-            ]
-          },
-          "action": {
-            "type": "ACTION",
-            "shape": [
-              7
-            ]
-          }
-        },
-        "norm_map": {
-          "VISUAL": "MEAN_STD",
-          "STATE": "MIN_MAX",
-          "ACTION": "MIN_MAX"
-        }
-      },
-      "state_file": "policy_preprocessor_step_4_normalizer_processor.safetensors"
-    }
-  ]
-}

checkpoints/010000/pretrained_model/policy_preprocessor_step_4_normalizer_processor.safetensors

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1483261062d5f6b8a80141ebb343170a30f06f9cda00c5b433b378d54610753c
-size 7616

checkpoints/010000/pretrained_model/train_config.json

@@ -1,257 +0,0 @@
-{
-    "dataset": {
-        "repo_id": "Bigenlight/cube_in_cup_lerobot_v3",
-        "root": "/home/junhyeong/workspace/youngwoong/cube_flow_matching/dataset",
-        "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",
-        "return_uint8": false,
-        "depth_output_unit": "mm",
-        "streaming": false,
-        "eval_split": 0.13
-    },
-    "env": null,
-    "policy": {
-        "type": "multi_task_dit",
-        "n_obs_steps": 2,
-        "input_features": {
-            "observation.state": {
-                "type": "STATE",
-                "shape": [
-                    7
-                ]
-            },
-            "observation.images.cam1": {
-                "type": "VISUAL",
-                "shape": [
-                    3,
-                    720,
-                    1280
-                ]
-            },
-            "observation.images.cam2": {
-                "type": "VISUAL",
-                "shape": [
-                    3,
-                    720,
-                    1280
-                ]
-            }
-        },
-        "output_features": {
-            "action": {
-                "type": "ACTION",
-                "shape": [
-                    7
-                ]
-            }
-        },
-        "device": "cuda",
-        "use_amp": false,
-        "use_peft": false,
-        "push_to_hub": false,
-        "repo_id": null,
-        "private": null,
-        "tags": null,
-        "license": null,
-        "pretrained_path": null,
-        "pretrained_revision": null,
-        "horizon": 32,
-        "n_action_steps": 24,
-        "objective": "flow_matching",
-        "noise_scheduler_type": "DDPM",
-        "num_train_timesteps": 100,
-        "beta_schedule": "squaredcos_cap_v2",
-        "beta_start": 0.0001,
-        "beta_end": 0.02,
-        "prediction_type": "epsilon",
-        "clip_sample": true,
-        "clip_sample_range": 1.0,
-        "num_inference_steps": null,
-        "sigma_min": 0.0,
-        "num_integration_steps": 100,
-        "integration_method": "euler",
-        "timestep_sampling_strategy": "beta",
-        "timestep_sampling_s": 0.999,
-        "timestep_sampling_alpha": 1.5,
-        "timestep_sampling_beta": 1.0,
-        "hidden_dim": 512,
-        "num_layers": 6,
-        "num_heads": 8,
-        "dropout": 0.1,
-        "use_positional_encoding": false,
-        "timestep_embed_dim": 256,
-        "use_rope": true,
-        "rope_base": 10000.0,
-        "vision_encoder_name": "openai/clip-vit-base-patch16",
-        "use_separate_rgb_encoder_per_camera": false,
-        "vision_encoder_lr_multiplier": 0.1,
-        "image_resize_shape": [
-            224,
-            224
-        ],
-        "image_crop_shape": [
-            224,
-            224
-        ],
-        "image_crop_is_random": true,
-        "text_encoder_name": "openai/clip-vit-base-patch16",
-        "tokenizer_max_length": 77,
-        "tokenizer_padding": "max_length",
-        "tokenizer_padding_side": "right",
-        "tokenizer_truncation": true,
-        "normalization_mapping": {
-            "VISUAL": "MEAN_STD",
-            "STATE": "MIN_MAX",
-            "ACTION": "MIN_MAX"
-        },
-        "optimizer_lr": 2e-05,
-        "optimizer_betas": [
-            0.95,
-            0.999
-        ],
-        "optimizer_eps": 1e-08,
-        "optimizer_weight_decay": 0.0,
-        "scheduler_name": "cosine",
-        "scheduler_warmup_steps": 0,
-        "do_mask_loss_for_padding": false,
-        "drop_n_last_frames": 7
-    },
-    "reward_model": null,
-    "output_dir": "/home/junhyeong/workspace/youngwoong/cube_flow_matching/training/outputs/flow_matching_cube_in_cup",
-    "job_name": "flow_matching_cube_in_cup",
-    "resume": false,
-    "seed": 1000,
-    "cudnn_deterministic": false,
-    "num_workers": 4,
-    "batch_size": 8,
-    "prefetch_factor": 4,
-    "persistent_workers": true,
-    "steps": 100000,
-    "env_eval_freq": 20000,
-    "log_freq": 200,
-    "eval_steps": 2000,
-    "max_eval_samples": 0,
-    "tolerance_s": 0.0001,
-    "save_checkpoint": true,
-    "save_freq": 10000,
-    "use_policy_training_preset": true,
-    "optimizer": {
-        "type": "adam",
-        "lr": 2e-05,
-        "weight_decay": 0.0,
-        "grad_clip_norm": 10.0,
-        "betas": [
-            0.95,
-            0.999
-        ],
-        "eps": 1e-08
-    },
-    "scheduler": {
-        "type": "diffuser",
-        "num_warmup_steps": 0,
-        "name": "cosine"
-    },
-    "eval": {
-        "n_episodes": 50,
-        "batch_size": 44,
-        "use_async_envs": true,
-        "recording": false,
-        "recording_repo_id": null,
-        "recording_private": false
-    },
-    "wandb": {
-        "enable": false,
-        "disable_artifact": false,
-        "project": "lerobot",
-        "entity": null,
-        "notes": null,
-        "run_id": null,
-        "mode": null,
-        "add_tags": true
-    },
-    "peft": null,
-    "job": {
-        "target": null,
-        "image": "huggingface/lerobot-gpu:latest",
-        "timeout": "2d",
-        "detach": false,
-        "tags": []
-    },
-    "save_checkpoint_to_hub": false,
-    "sample_weighting": null,
-    "rename_map": {},
-    "checkpoint_path": null
-}