diff --git a/climax/core/utils.py b/climax/core/utils.py index b619e2cb215c8134dd62b1547a0ada6d549fce4b..1d0e71b9fa2f05fc93ad73eb198623722d40aa37 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),