fusilli.utils.csv_loss_plotter

Plotting the loss from a CSV file generated by the training process when logging is set to False. (If set to True, then a weights and biases logger is used.)

Functions

plot_loss_curve(figures_path, logger[, show])

Plots the loss curve from a CSV file generated by the training process.

plot_loss_curve(figures_path, logger, show=False)[source]

Plots the loss curve from a CSV file generated by the training process. The figure is saved to output_paths[“figures”]/losses if show is False, otherwise it is shown with plt.show().

Parameters:

figures_path (str) – Path to the directory where the figure will be saved.
logger (pytorch lightning logger) – Logger that was used to log the training process. CSVLogger.
show (bool) – If True, show the plot. If False, save the plot to a file. Default: False.

Return type:

None