multistage.plot_loss

multistage.plot_loss(checkpoint_path, figname, title='Loss')Source

Plot loss history from specified path.

Parameters:

• checkpoint_path (str or list[str]) – Path to the directory containing checkpoints.

• figname (str) – Filename to save the plot.

• title (str) – Title for plot.

Returns:

fig, ax – Matplotlib fig, ax.