From 50232ca4bb775a1cfd32a0bb5da3aff2db20118e Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 21 Oct 2021 12:53:23 +0200
Subject: [PATCH] Improved plotting for loss.
---
climax/core/utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/climax/core/utils.py b/climax/core/utils.py
index b619e2c..1d0e71b 100644
--- a/climax/core/utils.py
+++ b/climax/core/utils.py
@@ -167,7 +167,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5, palette='mako'):
# plot training and validation loss
colors = sns.color_palette(palette, n_colors=2)
for (k, v), c, marker, ax in zip(rm.items(), colors, markers, axes):
- ax.plot(v, 'o', color=c, markevery=marker)
+ ax.plot(v, 'o', ls='-', color=c, markevery=marker)
# x axis limits
axes[0].set_xticks(np.arange(0, ntbatches * (epochs[-1] + 1),
--
GitLab