lightningdot
copied
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
54 lines
2.0 KiB
54 lines
2.0 KiB
2 years ago
|
{
|
||
|
|
"train_txt_db": ["/db/itm_coco_train_base-cased.db/",
|
||
|
|
"/db/itm_coco_restval_base-cased.db"],
|
||
|
|
"train_img_dir": ["/img/coco_train2014/",
|
||
|
|
"/img/coco_val2014/"],
|
||
|
|
"train_neg_sample_p": 0.5,
|
||
|
|
"neg_sample_from": "i",
|
||
|
|
"eval_method": "rank",
|
||
|
|
"val_txt_db": ["/db/itm_coco_val_1k_0_base-cased.db/",
|
||
|
|
"/db/itm_coco_val_1k_1_base-cased.db/",
|
||
|
|
"/db/itm_coco_val_1k_2_base-cased.db/",
|
||
|
|
"/db/itm_coco_val_1k_3_base-cased.db/",
|
||
|
|
"/db/itm_coco_val_1k_4_base-cased.db/"],
|
||
|
|
"val_img_dir": ["/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/"],
|
||
|
|
"test_txt_db": ["/db/itm_coco_test_1k_0_base-cased.db/",
|
||
|
|
"/db/itm_coco_test_1k_1_base-cased.db/",
|
||
|
|
"/db/itm_coco_test_1k_2_base-cased.db/",
|
||
|
|
"/db/itm_coco_test_1k_3_base-cased.db/",
|
||
|
|
"/db/itm_coco_test_1k_4_base-cased.db/"],
|
||
|
|
"test_img_dir": ["/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/",
|
||
|
|
"/img/coco_val2014/"],
|
||
|
|
"checkpoint": "/pretrain/mlm_caption_bert-base.pt",
|
||
|
|
"cut_bert": -1,
|
||
|
|
"output_dir": "/storage/itm_tr/coco_bert_base-mlm_caption-corrected_img_bb_num-w_train_rv-step_40000",
|
||
|
|
"max_txt_len": 60,
|
||
|
|
"conf_th": 0.2,
|
||
|
|
"max_bb": 100,
|
||
|
|
"min_bb": 10,
|
||
|
|
"num_bb": 36,
|
||
|
|
"train_batch_size": 2048,
|
||
|
|
"val_batch_size": 4096,
|
||
|
|
"val_minibatch_size":400,
|
||
|
|
"test_minibatch_size":300,
|
||
|
|
"gradient_accumulation_steps": 8,
|
||
|
|
"learning_rate": 0.001,
|
||
|
|
"valid_steps": 1000,
|
||
|
|
"num_train_steps": 40000,
|
||
|
|
"optim": "adamax",
|
||
|
|
"decay": "linear",
|
||
|
|
"dropout": 0.1,
|
||
|
|
"falseweight_decay": 0.01,
|
||
|
|
"grad_norm": 2.0,
|
||
|
|
"warmup_steps": 1000,
|
||
|
|
"seed": 42,
|
||
|
|
"fp16": true
|
||
|
|
}
|