File size: 973 Bytes
5633c2e | 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 | {
"dataset": "/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered",
"checkpoint_in": "/workspace/models/pi05_base_pytorch",
"checkpoint_out": "/workspace/models/pi05_arx_ordered",
"norm_stats": "/workspace/datasets/RoboPref_dataset/stacking_blocks_ordered/norm_stats_arx_l5.json",
"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",
"embodiment": "arx_l5",
"git": "6993e1c1ff942e2857950bba5cb76ac6015520a7",
"train_samples": 183984
} |