train-hm-base-triple-hpc_0_imgcap.json

{
    "root_path": "./",
    "dataset_path": "../data",
    "precomputed_boxes": {
        "train": "data_train_d2_10-100_vg.tsv",
        "dev": "data_dev_seen_unseen_d2_10-100_vg.tsv",
        "test": "data_test_seen_d2_10-100_vg.tsv"
    },
    "captions_file": "imgcap/hm_caption.csv",
    "paraphrased_file":"textaug/data_test_paraphrased_nlpaug_bert-base-uncased.jsonl",
    "train_image_set": "train+dev",
    "val_image_set": "dev",
    "test_image_set": "test",
    "checkpoint": "../UNITER/storage/pretrained/uniter-base.pt",
    "model_config": "./config/uniter-base.json",
    "model": "triple",
    "use_img_type": false,
    "output_dir": "../model_asset/output_triple_dev_seen_unseen_0_imgcap/hm/base",
    "max_txt_len": 100,
    "conf_th": 0.2,
    "max_bb": 100,
    "min_bb": 10,
    "num_bb": 36,
    "train_batch_size": 3328,
    "val_batch_size": 2496,
    "gradient_accumulation_steps": 2,
    "learning_rate": 2e-05,
    "valid_steps": 285,
    "num_train_steps": 3420,
    "optim": "adamw",
    "betas": [
        0.9,
        0.98
    ],
    "dropout": 0.2,
    "weight_decay": 0.01,
    "grad_norm": 2.0,
    "warmup_steps": 285,
    "seed": 0,
    "fp16": true,
    "n_workers": 4,
    "pin_mem": true
}