{ "dataset": { "repo_id": "SieloRobotics/rebot_pick_place_ee", "repo_type": "dataset", "root": null, "episodes": null, "exclude_episodes": [ 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 ], "image_transforms": { "enable": false, "backend": "dataloader", "gpu_compile": true, "gpu_chunk_size": 32, "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": true, "depth_output_unit": "mm", "streaming": false, "streaming_episode_pool_size": null, "streaming_sampling_strategy": "remaining", "streaming_prefetch_episodes": 8, "streaming_byte_budget_gb": 8.0, "streaming_decode_threads": 2, "streaming_decoded_queue_size": 8, "streaming_native_http_connections": null, "streaming_native_http_subranges": 1, "video_decoder_cache_size": null, "eval_split": 0.02 }, "env": null, "policy": { "type": "multi_task_dit", "n_obs_steps": 2, "input_features": { "observation.images.wrist": { "type": "VISUAL", "shape": [ 3, 480, 640 ] }, "observation.images.top": { "type": "VISUAL", "shape": [ 3, 480, 640 ] }, "observation.state": { "type": "STATE", "shape": [ 10 ] } }, "output_features": { "action": { "type": "ACTION", "shape": [ 10 ] } }, "device": "cuda", "dtype": null, "use_amp": false, "use_peft": false, "push_to_hub": true, "repo_id": "SieloRobotics/dit_ee_cubes_ep80", "private": false, "tags": null, "license": null, "pretrained_path": null, "pretrained_revision": null, "horizon": 32, "n_action_steps": 8, "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": 10, "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": true, "vision_encoder_lr_multiplier": 0.1, "image_resize_shape": [ 240, 240 ], "image_crop_shape": [ 224, 224 ], "image_crop_is_random": true, "compile_model": true, "compile_mode": null, "use_text_conditioning": false, "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": "QUANTILES", "ACTION": "QUANTILES" }, "use_relative_ee_actions": true, "action_feature_names": [ "ee.x", "ee.y", "ee.z", "ee.r6d_0", "ee.r6d_1", "ee.r6d_2", "ee.r6d_3", "ee.r6d_4", "ee.r6d_5", "gripper.pos" ], "optimizer_lr": 0.0001, "optimizer_betas": [ 0.95, 0.999 ], "optimizer_eps": 1e-08, "optimizer_weight_decay": 0.0, "optimizer_grad_clip_norm": 1.0, "scheduler_name": "cosine", "scheduler_warmup_steps": 800, "do_mask_loss_for_padding": false, "drop_n_last_frames": 23 }, "reward_model": null, "output_dir": "/content/outputs/dit_ee_cubes_ep80_run_20261020_204725", "job_name": "dit_ee_cubes_ep80", "resume": false, "seed": 1000, "cudnn_deterministic": false, "num_workers": 12, "batch_size": 256, "prefetch_factor": 2, "persistent_workers": true, "dataloader_multiprocessing_context": "spawn", "steps": 40000, "env_eval_freq": 20000, "log_freq": 100, "eval_steps": 2000, "max_eval_samples": 0, "tolerance_s": 0.0001, "save_checkpoint": true, "save_freq": 2000, "checkpoint_format": "safetensors", "use_policy_training_preset": true, "optimizer": { "type": "adam", "lr": 0.0001, "weight_decay": 0.0, "grad_clip_norm": 1.0, "betas": [ 0.95, 0.999 ], "eps": 1e-08 }, "scheduler": { "type": "diffuser", "num_warmup_steps": 800, "name": "cosine" }, "parallelism": { "dp_replicate": 1, "dp_shard": 1, "context_parallel": { "ring_degree": 1, "ulysses_degree": 1 }, "cfg_parallel": 1 }, "accelerator": { "mixed_precision": "bf16", "gradient_accumulation": { "steps": 1 }, "grad_scaler": { "init_scale": 65536.0, "growth_factor": 2.0, "backoff_factor": 0.5, "growth_interval": 2000 }, "fsdp": { "reshard_after_forward": true, "wrap_modules": null, "min_num_params": null, "cpu_offload": false, "ignored_modules": null }, "ddp": { "find_unused_parameters": true, "gradient_as_bucket_view": false, "static_graph": false }, "compile": { "enabled": false, "backend": "inductor", "mode": null, "regional": true }, "activation_checkpointing": { "mode": "none" } }, "eval": { "n_episodes": 50, "batch_size": 33, "use_async_envs": true, "recording": false, "recording_repo_id": null, "recording_private": false }, "ema": { "enable": true, "decay": null, "update_after_step": 0, "inv_gamma": 1.0, "power": 0.75, "min_decay": 0.0, "max_decay": 0.9999, "use_for_eval": true }, "wandb": { "enable": true, "disable_artifact": true, "project": "dit_ee_cubes", "entity": null, "notes": "episodes 80-511 (0-79 excluded), otherwise as dit_ee_cubes", "run_id": "dzzxm1zm", "resume": null, "mode": null, "console": "wrap", "console_multipart": false, "console_chunk_max_seconds": 0, "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 }