feat: update policy

emma-heriot-watt · Nov 30, 2023 · d7c2d20 · d7c2d20
1 parent 903a4af
commit d7c2d20
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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: