Added output of train and test/validation loss at the end of each epoch

2 jobs for develop in 33 seconds (queued for 1 second)