all debug and train running

lib/utils/logging.py

@@ -71,5 +71,25 @@ class Logger(LightningLoggerBase):
         self.testtubelogger.log_metrics(metrics, step=step)
         pass
 
+    def close(self):
+        self.testtubelogger.close()
+        self.neptunelogger.close()
+
     def log_config_as_ini(self):
         self.config.write(self.log_dir)
+
+    def save(self):
+        self.testtubelogger.save()
+        self.neptunelogger.save()
+
+    def finalize(self, status):
+        self.testtubelogger.finalize()
+        self.neptunelogger.finalize()
+        self.log_config_as_ini()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.finalize('success')
+        pass