Validation fix since we don't have multi-GPU DataParallel support yet

pull/1239/head
Ross Wightman 3 years ago
parent 59a3409182
commit 07693f81b0

@ -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)

Loading…
Cancel
Save