Plotting code

[kogolobo]

We need to be able to plot the loss value for different optimizers and stepping schemes. We will extend the trainer to produce the training curve plot.