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
43 lines
1.3 KiB
43 lines
1.3 KiB
{
|
|
"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
|
|
}
|