{ "txt_model_type": "bert-base", "txt_model_config": "bert-base-cased", "img_model_type": "uniter-base", "img_model_config": "./config/img_base.json", "img_checkpoint": "./data/model/uniter-base.pt", "itm_global_file":"./data/meta/coco_meta.json", "train_batch_size": 64, "val_batch_size": 256, "gradient_accumulation_steps": 1, "learning_rate": 2e-05, "warmup_steps": 100, "valid_steps": 500, "num_train_epochs": 20, "seed": 42, "output_dir": "/storage/debug_coco", "max_txt_len": 60, "conf_th": 0.2, "max_bb": 100, "min_bb": 10, "num_bb": 36, "project_dim": 768, "train_txt_dbs": [ "./data/db/itm_coco_train_base-cased.db", "./data/db/itm_coco_restval_base-cased.db" ], "train_img_dbs": [ "./data/img/coco_train2014/", "./data/img/coco_val2014" ], "val_txt_db": "./data/db/itm_coco_val_base-cased.db", "val_img_db": "./data/img/coco_val2014/", "test_txt_db": "./data/db/itm_coco_test_base-cased.db", "test_img_db": "./data/img/coco_val2014/", "project_name": "itm-debug", "num_hard_negatives": 0, "hard_negatives_sampling": "none", "inf_minibatch_size": 0, "n_workers": 0, "fp16": true, "compressed_db": false, "pin_mem": true }