Fixed index out of range in case of resume

2024-01-11 14:47:56 +01:00 · 2024-01-11 14:47:56 +01:00 · 8c663c4b86
parent 36449617ff
commit 8c663c4b86
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -912,7 +912,7 @@ def main():

    results = {'all': results}
    if best_metric is not None:
-        results['best'] = results['all'][best_epoch]
+        results['best'] = results['all'][best_epoch - start_epoch]
        _logger.info('*** Best metric: {0} (epoch {1})'.format(best_metric, best_epoch))
    print(f'--result\n{json.dumps(results, indent=4)}')