|
|
@@ -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()
|