File size: 1,054 Bytes
de183da | 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 34 35 36 37 38 39 40 41 | {
"batch_size": 2,
"accum_iter": 4,
"epochs": 2,
"warmup_epochs": 0.001,
"lr": 2e-05,
"min_lr": 2e-05,
"wd": 0.1,
"clip_grad": 4.0,
"init_from": "/mnt/petrelfs/xinyi/Lumina-DiMOO/output/Lumina-DiMOO-8B-Joint-Supervised-Fine-Tuning",
"data_config": "/mnt/petrelfs/xinyi/Lumina-DiMOO/configs/data.yaml",
"cache_ann_on_disk": true,
"length_clustering": true,
"num_workers": 16,
"pin_mem": true,
"seed": 0,
"output_dir": "output/Lumina-DiMOO-8B-Joint-Supervised-Fine-Tuning-Final",
"save_interval": 1,
"save_iteration_interval": 200,
"only_save_trainable": false,
"ckpt_max_keep": 2,
"auto_resume": true,
"resume_path": null,
"model_parallel_size": 1,
"data_parallel": "sdp",
"precision": "bf16",
"grad_precision": "fp32",
"checkpointing": false,
"max_seq_len": 5120,
"mask_image_logits": false,
"dropout": 0.05,
"z_loss_weight": 1e-05,
"model_size": "7B",
"world_size": 48,
"rank": 0,
"gpu": 0,
"local_rank": 0,
"dist_url": "env://",
"distributed": true,
"dist_backend": "nccl"
} |