Update on_train_end callback (#7716)
This commit is contained in:
1 parent
1d0a0944c1
commit
aa7a0e91b1
2 files changed
+1
-1
No files matched your search
@@ -164,6 +164,7 @@ class Loggers():
|
||||
plot_results(file=self.save_dir / 'results.csv') # save results.png
|
||||
files = ['results.png', 'confusion_matrix.png', *(f'{x}_curve.png' for x in ('F1', 'PR', 'P', 'R'))]
|
||||
files = [(self.save_dir / f) for f in files if (self.save_dir / f).exists()] # filter
|
||||
self.logger.info(f"Results saved to {colorstr('bold', self.save_dir)}")
|
||||
|
||||
if self.tb:
|
||||
for f in files:
|
||||
|
||||
Reference in new issue
Block a user