mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Fix missing model key in bulk validate results on error
This commit is contained in:
parent
b3a83b81d6
commit
b1752eefb5
@ -412,6 +412,7 @@ def _try_run(args, initial_batch_size):
|
||||
break
|
||||
batch_size = decay_batch_step(batch_size)
|
||||
_logger.warning(f'Reducing batch size to {batch_size} for retry.')
|
||||
results['model'] = args.model
|
||||
results['error'] = error_str
|
||||
_logger.error(f'{args.model} failed to validate ({error_str}).')
|
||||
return results
|
||||
|
Loading…
x
Reference in New Issue
Block a user