Skip to content

Commit b1752ee

Browse files
committed
Fix missing model key in bulk validate results on error
1 parent b3a83b8 commit b1752ee

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

validate.py

+1
Original file line numberDiff line numberDiff line change
@@ -412,6 +412,7 @@ def _try_run(args, initial_batch_size):
412412
break
413413
batch_size = decay_batch_step(batch_size)
414414
_logger.warning(f'Reducing batch size to {batch_size} for retry.')
415+
results['model'] = args.model
415416
results['error'] = error_str
416417
_logger.error(f'{args.model} failed to validate ({error_str}).')
417418
return results

0 commit comments

Comments
 (0)