From 425e124d837524d8c5c076481eabc099514a201f Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Tue, 21 Dec 2021 16:41:02 +0000 Subject: [PATCH] fix bs --- ppstructure/vqa/vqa_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppstructure/vqa/vqa_utils.py b/ppstructure/vqa/vqa_utils.py index 8f5ff2ef..b9f2edc8 100644 --- a/ppstructure/vqa/vqa_utils.py +++ b/ppstructure/vqa/vqa_utils.py @@ -366,9 +366,9 @@ def parse_args(): parser.add_argument("--max_seq_length", default=512, type=int,) parser.add_argument("--evaluate_during_training", action="store_true",) parser.add_argument("--num_workers", default=8, type=int,) - parser.add_argument("--per_gpu_train_batch_size", default=1, + parser.add_argument("--per_gpu_train_batch_size", default=8, type=int, help="Batch size per GPU/CPU for training.",) - parser.add_argument("--per_gpu_eval_batch_size", default=1, + parser.add_argument("--per_gpu_eval_batch_size", default=8, type=int, help="Batch size per GPU/CPU for eval.",) parser.add_argument("--learning_rate", default=5e-5, type=float, help="The initial learning rate for Adam.",) -- GitLab