{
  "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
}