Improved plotting for loss.

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),