{
  "batch_size": 32,
  "epochs": 800,
  "model": "mar_huge",
  "img_size": 256,
  "vae_path": "/taijifs_cq6/share_304079515/hunyuan/tianbali/wpit_runs_20261014/models/wpit/ckpt-last-90%WPIT.pth",
  "vae_embed_dim": 64,
  "vae_stride": 16,
  "patch_size": 1,
  "num_iter": 256,
  "num_images": 50000,
  "cfg": 1.0,
  "cfg_schedule": "linear",
  "label_drop_prob": 0.1,
  "eval_freq": 20,
  "save_last_freq": 50,
  "online_eval": false,
  "evaluate": false,
  "eval_bsz": 64,
  "sample_labels": "",
  "skip_metrics": false,
  "eval_ema": true,
  "weight_decay": 0.02,
  "grad_checkpointing": true,
  "lr": 0.0004,
  "blr": 0.0001,
  "min_lr": 0.0,
  "lr_schedule": "constant",
  "warmup_epochs": 100,
  "ema_rate": 0.999,
  "mask_ratio_min": 0.7,
  "grad_clip": 3.0,
  "attn_dropout": 0.1,
  "proj_dropout": 0.1,
  "buffer_size": 64,
  "diffloss_d": 12,
  "diffloss_w": 1536,
  "num_sampling_steps": "100",
  "diffusion_batch_mul": 4,
  "diff_clip_denoised": true,
  "temperature": 1.0,
  "data_path": "./data/imagenet",
  "class_num": 1000,
  "code_path": "./data/imagenet",
  "dataset": "imagenet_code",
  "output_dir": "/taijifs_cq6/share_304079515/hunyuan/tianbali/wpit_runs_20261014/checkpoints/mar_huge_imagenet_ep800",
  "log_dir": "/taijifs_cq6/share_304079515/hunyuan/tianbali/wpit_runs_20261014/checkpoints/mar_huge_imagenet_ep800",
  "device": "cuda",
  "seed": 1,
  "resume": "",
  "resume_epoch": null,
  "start_epoch": 0,
  "num_workers": 4,
  "pin_mem": true,
  "world_size": 32,
  "local_rank": -1,
  "dist_on_itp": false,
  "dist_url": "env://",
  "use_cached": false,
  "cached_path": "/taijifs_cq6/share_304079515/hunyuan/tianbali/wpit_runs_20261014/codes/ckpt-last-90pctWPIT/imagenet_code_c2i_crop0",
  "feature_stats": "",
  "fixed_aug_index": 0,
  "use_wandb": false,
  "wandb_project": null,
  "unconditional": false,
  "feature_extractor_path": "./fid_stats/pt_inception-2015-12-05-6726825d.pth",
  "exp_name": "test",
  "fid_method": "fid_stats",
  "fid_stats_file": "./fid_stats/adm_in256_stats.npz",
  "fid_reference_dir": "/root/data/imagenet_reference_batch/imagenet_256_val",
  "cleanup_eval_folder": false,
  "rank": 0,
  "gpu": 0,
  "distributed": true,
  "dist_backend": "nccl"
}