{ "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 }