deepof.model_utils.plot_lr_vs_loss

deepof.model_utils.plot_lr_vs_loss(rates, losses)

Plot learning rate versus the loss function of the model.

Parameters:
  • rates (np.ndarray) – array containing the learning rates to plot in the x-axis

  • losses (np.ndarray) – array containing the losses to plot in the y-axis