Hi!, is it possible to get the loss across every epoch?