From 288838b41979be40eec181ce69cb3ba80597b01d Mon Sep 17 00:00:00 2001 From: Junnan Li Date: Fri, 16 Jul 2021 08:47:00 +0800 Subject: [PATCH] Update Retrieval.py --- Retrieval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Retrieval.py b/Retrieval.py index eb4df4e..f5db5b9 100644 --- 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) \ No newline at end of file + main(args, config)