Update Retrieval.py

Author: Junnan Li

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)