| { |
| "args": { |
| "dataset_dir": "datasets/hepha_act_200_simple_drawer_5", |
| "output_dir": "runs/hepha_diffusion_200_simple_drawer_5_obs4", |
| "epochs": 50, |
| "batch_size": 16, |
| "num_workers": 4, |
| "lr": 0.0001, |
| "weight_decay": 1e-06, |
| "chunk_size": 50, |
| "image_height": 240, |
| "image_width": 320, |
| "obs_feature_dim": 512, |
| "n_obs_steps": 4, |
| "n_action_steps": null, |
| "diffusion_step_embed_dim": 256, |
| "down_dims": [ |
| 256, |
| 512, |
| 1024 |
| ], |
| "kernel_size": 5, |
| "n_groups": 8, |
| "cond_predict_scale": true, |
| "diffusion_steps": 100, |
| "num_inference_steps": 100, |
| "beta_schedule": "squaredcos_cap_v2", |
| "prediction_type": "epsilon", |
| "backbone": "resnet18", |
| "lr_backbone": 1e-05, |
| "no_pretrained_backbone": false, |
| "dropout": 0.1, |
| "val_ratio": 0.1, |
| "val_every_steps": 4000, |
| "seed": 42, |
| "log_every": 50, |
| "save_every": 0, |
| "save_every_steps": 4000, |
| "resume": null, |
| "amp": true, |
| "device": "cuda", |
| "wandb_project": "hepha-diffusion", |
| "wandb_entity": null, |
| "wandb_name": "hepha-diffusion-200-simple-drawer-5-obs4", |
| "wandb_tags": null, |
| "wandb_mode": null, |
| "wandb_log_checkpoints": false, |
| "hf_model_repo_id": "tmeynier/hepha-diffusion-200-simple-drawer-5-obs4", |
| "hf_revision": null, |
| "hf_private": true, |
| "hf_upload_every": null, |
| "hf_upload_every_steps": 4000, |
| "hf_upload_fail_training": false |
| }, |
| "model_config": { |
| "state_dim": 16, |
| "action_dim": 15, |
| "chunk_size": 50, |
| "horizon": 50, |
| "n_obs_steps": 4, |
| "n_action_steps": 50, |
| "obs_feature_dim": 512, |
| "diffusion_step_embed_dim": 256, |
| "down_dims": [ |
| 256, |
| 512, |
| 1024 |
| ], |
| "kernel_size": 5, |
| "n_groups": 8, |
| "cond_predict_scale": true, |
| "diffusion_steps": 100, |
| "num_inference_steps": 100, |
| "beta_start": 0.0001, |
| "beta_end": 0.02, |
| "beta_schedule": "squaredcos_cap_v2", |
| "prediction_type": "epsilon", |
| "clip_sample": true, |
| "dropout": 0.1, |
| "backbone": "resnet18", |
| "pretrained_backbone": true |
| }, |
| "dataset_info": { |
| "dataset_dir": "datasets/hepha_act_200_simple_drawer_5", |
| "total_frames": 253867, |
| "episodes": 200, |
| "state_dim": 16, |
| "action_dim": 15 |
| } |
| } |
|
|
