- Change descriptors in trainer fucntion (currently images and labels) - Include flag to plot losses (default should be false)