| { | |
| "dataset": "/workspace/datasets/RoboPref_dataset_v3/stacking_blocks_ordered", | |
| "checkpoint_in": "/workspace/models/pi05_base_pytorch", | |
| "checkpoint_out": "/workspace/models/pi05_arx_ordered_v3d", | |
| "norm_stats": "/workspace/datasets/RoboPref_dataset_v3/stacking_blocks_ordered/norm_stats_arx_l5_3cam.json", | |
| "embodiment": "arx_l5_3cam", | |
| "tokenizer": "/root/.cache/openpi/big_vision/paligemma_tokenizer.model", | |
| "steps": 15000, | |
| "micro_batch": 32, | |
| "accum": 1, | |
| "lr": 2.5e-05, | |
| "warmup": 1000, | |
| "final_lr_frac": 0.1, | |
| "grad_clip": 1.0, | |
| "num_workers": 12, | |
| "seed": 0, | |
| "device": "cuda", | |
| "offload_optim": false, | |
| "ema": null, | |
| "resume": false, | |
| "overfit_episode": null, | |
| "log_every": 20, | |
| "val_every": 1000, | |
| "action_val_every": 5000, | |
| "save_every": 2500, | |
| "val_samples": 512, | |
| "action_val_samples": 64, | |
| "wandb_project": "robopref", | |
| "wandb_entity": null, | |
| "wandb_mode": "online", | |
| "git": "95b6b5e93d3c6d438fc29ddea347e6f47093ab1d", | |
| "train_samples": 329138 | |
| } |