Update Retrieval.py

SuJQ0722 · Jul 16, 2021 · 288838b · 288838b
1 parent 750a2a8
commit 288838b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Retrieval.py b/Retrieval.py
@@ -361,7 +361,7 @@ def main(args, config):
     parser.add_argument('--config', default='./configs/Retrieval_flickr.yaml')
     parser.add_argument('--output_dir', default='output/Retrieval_flickr')        
     parser.add_argument('--checkpoint', default='')   
-    parser.add_argument('--text_encoder', default='bert-large-uncased')
+    parser.add_argument('--text_encoder', default='bert-base-uncased')
     parser.add_argument('--evaluate', action='store_true')
     parser.add_argument('--device', default='cuda')
     parser.add_argument('--seed', default=42, type=int)
@@ -376,4 +376,4 @@ def main(args, config):
 
     yaml.dump(config, open(os.path.join(args.output_dir, 'config.yaml'), 'w'))    
 
-    main(args, config)
+    main(args, config)