Add batch size log for accuracy mode
@@ -229,6 +229,7 @@ class Benchmark(object):
for i, res in enumerate(result_list):
logger.debug('Iteration {} result {}:'.format(i, res))
if mode == 'accuracy':
+ logger.info('Batch size = {}'.format(batch_size))
logger.info('Accuracy is {:.4f}'.format(acc))
elif mode == 'performance':
logger.info('Batch size = {}'.format(batch_size))
