diff options
| author | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-02 18:46:13 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-02 18:46:13 +0200 |
| commit | 45b67cbadd51d81bd55e759a07f1e1a5fbf5d5c4 (patch) | |
| tree | 6bd0b7342bbb0d850d037a5a34095153e09fa40a | |
| parent | 25c01ba5092994f5156922e4873281b0a2e69b15 (diff) | |
Write numbers instead of tensors
| -rw-r--r-- | src/experiment.py | 4 |
1 files 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): |