Add log axis option for plotting losses?

I agree, please help by upvoting my feature request: Improving plots, especially `plot_loss()`