Training running

2020-06-08 10:23:12 +02:00
parent 27ae8467fc
commit 821b2d1961
3 changed files with 11 additions and 8 deletions
--- a/utils/module_mixins.py
+++ b/utils/module_mixins.py
@ -213,6 +213,8 @@ class BaseValMixin:
        disp.plot(include_values=True)
        self.logger.log_image('Confusion Matrix', image=disp.figure_, step=self.current_epoch)

+        plt.close('all')
+
        return summary_dict