From 0639d9a591b175b519b40aa2ce70aae11a0d6708 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 2 Sep 2021 14:44:53 -0700 Subject: [PATCH] Fix updated validation_batch_size fallback --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 929948d8..3943c7d0 100755 --- a/train.py +++ b/train.py @@ -539,7 +539,7 @@ def main(): loader_eval = create_loader( dataset_eval, input_size=data_config['input_size'], - batch_size=args.validation_batch_size, + batch_size=args.validation_batch_size or args.batch_size, is_training=False, use_prefetcher=args.prefetcher, interpolation=data_config['interpolation'],