logo
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

25 lines
502 B

train_file: ['data/nlvr_train.json']
val_file: ['data/nlvr_dev.json']
test_file: ['data/nlvr_test.json']
image_root: '/export/share/datasets/vision/NLVR2/'
image_res: 384
batch_size: 16
bert_config: 'configs/config_bert.json'
alpha: 0.4
distill: True
warm_up: True
eval_ema: False
optimizer: {opt: adamW, lr: 2e-5, weight_decay: 0.02}
schedular: {sched: cosine, lr: 2e-5, epochs: 10, min_lr: 1e-6, decay_rate: 1, warmup_lr: 1e-5, warmup_epochs: 1, cooldown_epochs: 0}