Plotting train and test loss

1

Hi, trying to plot training and test loss in same graph, any advice?

2

You could use matplotlib.pylot to plot multiple graphs in the same plot.
Alternatively, PyTorch has also TensorBoard as well as Visdom support.

3

@ptrblck will try both. Thanks for help