diff --git a/climax/core/utils.py b/climax/core/utils.py index 3ec1118f2e6b6ce2dbb18c56405451ac11d8ffa4..485e1a80ff679e6288329240ecda740b84bbdf6d 100644 --- a/climax/core/utils.py +++ b/climax/core/utils.py @@ -179,6 +179,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5): yl_max, yl_min = (ceil_decimal(max_loss, decimal=1), floor_decimal(min_loss, decimal=1)) axes[0].set_ylim(yl_min, yl_max) + axes[0].set_yticks(np.arange(yl_min, yl_max + 0.2, 0.2), fontsize=14) # compute early stopping point if loss['valid_loss'].any():