From 07693f81b06cfdfac77230e4a0a16a70278842b3 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Wed, 10 Nov 2021 15:54:21 -0800
Subject: [PATCH] Validation fix since we don't have multi-GPU DataParallel
 support yet

---
 validate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/validate.py b/validate.py
index ad746ffc..24c5dc59 100755
--- a/validate.py
+++ b/validate.py
@@ -266,7 +266,7 @@ def main():
                 args.checkpoint = c
                 result = OrderedDict(model=args.model)
                 r = {}
-                while not r and batch_size >= args.num_gpu:
+                while not r and batch_size >= 1:
                     try:
                         args.batch_size = batch_size
                         print('Validating with batch size: %d' % args.batch_size)