{
  "batch_size": 1,
  "accum_iter": 8,
  "epochs": 2,
  "warmup_epochs": 0.01,
  "lr": 0.0001,
  "min_lr": 0.0001,
  "wd": 0.1,
  "clip_grad": 4.0,
  "init_from": "/mnt/petrelfs/xinyi/synbol_code/Lumina-mGPT-2.0-TPAMI/lumina_mgpt/output_final_tpami/7B-tpami/epoch0-iter15375",
  "data_config": "configs/data/data.yaml",
  "cache_ann_on_disk": true,
  "length_clustering": true,
  "num_workers": 8,
  "pin_mem": true,
  "seed": 0,
  "output_dir": "output_final_tpami/7B-tpami-v3-final",
  "save_interval": 1,
  "save_iteration_interval": 5000,
  "only_save_trainable": false,
  "ckpt_max_keep": 2,
  "auto_resume": true,
  "resume_path": null,
  "model_parallel_size": 1,
  "data_parallel": "fsdp",
  "precision": "bf16",
  "grad_precision": "fp32",
  "checkpointing": true,
  "max_seq_len": 10240,
  "mask_image_logits": false,
  "dropout": 0.05,
  "z_loss_weight": 1e-05,
  "model_size": "7B",
  "world_size": 128,
  "rank": 0,
  "gpu": 0,
  "local_rank": 0,
  "dist_url": "env://",
  "distributed": true,
  "dist_backend": "nccl"
}