From bb51614387029acfb64eb72df4411bba063497b5 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 12 May 2020 22:59:43 +0200
Subject: [PATCH] [engine.ssltrainer] Make logged variable names more
consistent
---
bob/ip/binseg/engine/ssltrainer.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/bob/ip/binseg/engine/ssltrainer.py b/bob/ip/binseg/engine/ssltrainer.py
index 6905a645..dcc50005 100644
--- a/bob/ip/binseg/engine/ssltrainer.py
+++ b/bob/ip/binseg/engine/ssltrainer.py
@@ -280,8 +280,8 @@ def run(
"eta",
"average_loss",
"median_loss",
- "median_labelled_loss",
- "median_unlabelled_loss",
+ "labelled_median_loss",
+ "unlabelled_median_loss",
"learning_rate",
)
if valid_loader is not None:
@@ -429,8 +429,8 @@ def run(
("eta", f"{datetime.timedelta(seconds=int(eta_seconds))}"),
("average_loss", f"{losses.avg:.6f}"),
("median_loss", f"{losses.median:.6f}"),
- ("median_labelled_loss", f"{labelled_loss.median:.6f}"),
- ("median_unlabelled_loss", f"{unlabelled_loss.median:.6f}"),
+ ("labelled_median_loss", f"{labelled_loss.median:.6f}"),
+ ("unlabelled_median_loss", f"{unlabelled_loss.median:.6f}"),
("learning_rate", f"{optimizer.param_groups[0]['lr']:.6f}"),
)
if valid_losses is not None:
--
GitLab