Przeglądaj źródła

Merge branch 'master' of triton.inf-cv.uni-jena.de:korsch/cvfinetune

Dimitri Korsch 5 lat temu
rodzic
commit
7f7917ad61
1 zmienionych plików z 12 dodań i 3 usunięć
  1. 12 3
      cvfinetune/training/trainer/base.py

+ 12 - 3
cvfinetune/training/trainer/base.py

@@ -166,9 +166,18 @@ class Trainer(T):
 		if init_eval:
 			logging.info("Evaluating initial model ...")
 			init_perf = self.evaluator(self)
-			logging.info("Initial accuracy: {val/main/accuracy:.3%} initial loss: {val/main/loss:.3f}".format(
-				**{key: float(value) for key, value in init_perf.items()}
-			))
+			values = {key: float(value) for key, value in init_perf.items()}
+
+			msg = []
+
+			if "val/main/accuracy" in values:
+				msg.append("Initial accuracy: {val/main/accuracy:.3%}".format(**values))
+
+			if "val/main/loss" in values:
+				msg.append("Initial loss: {val/main/loss:.3f}".format(**values))
+
+			logging.info(" ".join(msg))
+
 		if self._only_eval:
 			return
 		return super(Trainer, self).run()