{ "train_txt_dbs": ["/db/itm_flickr30k_val_base-cased.db"], "train_img_dbs": ["/img/flickr30k/"], "val_txt_db": "/db/itm_flickr30k_val_base-cased.db", "val_img_db": "/img/flickr30k/", "test_txt_db": "/db/itm_flickr30k_test_base-cased.db", "test_img_db": "/img/flickr30k/", "checkpoint": "/pretrain/uniter-base-iclr.pt", "model_config": "/src/config/uniter-base.json", "output_dir": "/debug/itm/flickr_default", "max_txt_len": 60, "conf_th": 0.2, "max_bb": 100, "min_bb": 10, "num_bb": 36, "train_batch_size": 20, "negative_size": 1, "hard_neg_size": 1, "hard_neg_pool_size": 20, "steps_per_hard_neg": 30, "inf_minibatch_size": 40, "gradient_accumulation_steps": 2, "learning_rate": 1e-05, "valid_steps": 40, "num_train_steps": 50, "optim": "adamw", "betas": [ 0.9, 0.98 ], "margin": 0.2, "dropout": 0.1, "weight_decay": 0.01, "grad_norm": 2.0, "warmup_steps": 1600, "seed": 42, "fp16": true, "n_workers": 0, "pin_mem": true }