eval written

lib/utils/logging.py

@@ -78,6 +78,14 @@ class Logger(LightningLoggerBase):
     def log_config_as_ini(self):
         self.config.write(self.log_dir)
 
+    def log_metric(self, metric_name, metric_value, **kwargs):
+        self.testtubelogger.log_metrics(dict(metric_name=metric_value))
+        self.neptunelogger.log_metric(metric_name, metric_value, **kwargs)
+
+    def log_image(self, name, image, **kwargs):
+        self.neptunelogger.log_image(name, image, **kwargs)
+        image.savefig(self.log_dir / name)
+
     def save(self):
         self.testtubelogger.save()
         self.neptunelogger.save()