Fix updated validation_batch_size fallback

2021-09-02 14:44:53 -07:00 · 2021-09-02 14:44:53 -07:00 · 0639d9a591
parent 5db057dca0
commit 0639d9a591
1 changed files with 1 additions and 1 deletions
--- 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'],