OFA_huge_NICE_captioning / train_args.json
calisolo's picture
Upload 3 files
4101489
raw
history blame
950 Bytes
{
"output_dir": "../drive/MyDrive/NICE/model/0422_train_encoder",
"model_name_or_path": "../drive/MyDrive/NICE/model/cap_best/OFA-huge-caption",
"train_caption_file": "./data/train_caption.jsonl",
"train_image_file": "../drive/MyDrive/NICE/train_image.tsv",
"test_caption_file": null,
"test_image_file": null,
"freeze_encoder": false,
"freeze_word_embed": false,
"num_train_epochs": 30,
"max_steps": -1,
"per_device_train_batch_size": 12,
"per_device_eval_batch_size": 256,
"learning_rate": 5e-05,
"max_seq_length": 150,
"logging_steps": 5000,
"save_steps": 5000,
"save_total_limit": 2,
"lr_scheduler_type": "cosine",
"warmup_steps": 2000,
"warmup_ratio": 0,
"gradient_accumulation_steps": 1,
"optim": "adamw_torch",
"seed": 69,
"fp16": false,
"no_cuda": false,
"dataloader_num_workers": 8,
"save_strategy": "steps",
"weight_decay": 0,
"max_grad_norm": 1.0,
"remove_unused_columns": false
}