diff --git a/configs/experiment/nlvr2_downstream.yaml b/configs/experiment/nlvr2_downstream.yaml index f13225a..1ae5a2c 100644 --- a/configs/experiment/nlvr2_downstream.yaml +++ b/configs/experiment/nlvr2_downstream.yaml @@ -52,9 +52,9 @@ datamodule: nlvr2_train_db_file: storage/db/nlvr2_fixtures/nlvr_train.db nlvr2_valid_db_file: storage/db/nlvr2_fixtures/nlvr_valid_seen.db nlvr2_test_db_file: storage/db/nlvr2_fixtures/nlvr_valid_unseen.db - train_batch_size: 100 - val_batch_size: 100 - num_workers: 0 + train_batch_size: 64 + val_batch_size: 128 + num_workers: 4 max_lang_tokens: 512 tokenizer_truncation_side: right diff --git a/configs/experiment/refcoco_downstream.yaml b/configs/experiment/refcoco_downstream.yaml index 65ce8f7..3427602 100644 --- a/configs/experiment/refcoco_downstream.yaml +++ b/configs/experiment/refcoco_downstream.yaml @@ -57,13 +57,13 @@ datamodule: refcoco_train_db_file: storage/db/refcoco_train.db refcoco_valid_db_file: storage/db/refcoco_valid.db refcoco_test_db_file: storage/db/refcoco_test.db - train_batch_size: 90 + train_batch_size: 32 val_batch_size: 100 - num_workers: 0 + num_workers: 4 max_lang_tokens: 64 tokenizer_truncation_side: right shuffle_objects: False - train_with_golden_bbox_prob: 1.0 + train_with_golden_bbox_prob: 0.5 callbacks: model_checkpoint: