mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix missing model key in bulk validate results on error

Browse Source
This commit is contained in:
Ross Wightman 2025-01-28 13:20:40 -08:00
parent b3a83b81d6
commit b1752eefb5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
validate.py
View File

@ -412,6 +412,7 @@ def _try_run(args, initial_batch_size):
break break
batch_size = decay_batch_step(batch_size) batch_size = decay_batch_step(batch_size)
_logger.warning(f'Reducing batch size to {batch_size} for retry.') _logger.warning(f'Reducing batch size to {batch_size} for retry.')
results['model'] = args.model
results['error'] = error_str results['error'] = error_str
_logger.error(f'{args.model} failed to validate ({error_str}).') _logger.error(f'{args.model} failed to validate ({error_str}).')
return results return results