From 45b67cbadd51d81bd55e759a07f1e1a5fbf5d5c4 Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Sun, 2 May 2021 18:46:13 +0200 Subject: Write numbers instead of tensors --- src/experiment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/experiment.py b/src/experiment.py index dd2033c..b508438 100644 --- a/src/experiment.py +++ b/src/experiment.py @@ -54,10 +54,10 @@ class ExperimentMetrics: self.test_accuracies = [] def add_train_loss(self, loss): - self.train_losses.append(loss) + self.train_losses.append(round(loss.tolist(), 3)) def add_test_metrics(self, loss, accuracy): - self.test_losses.append(loss) + self.test_losses.append(round(loss.tolist(), 3)) self.test_accuracies.append(accuracy) def get_dataframe(self): -- cgit v1.2.3