Decouple training_step return from logging
We should adopt a standard for all validation and training functions.
Maybe something like:
return {loss: loss, 
log: {},
prog: {}
}
Where log goes to self.experiment.
prog goes to the progress bar?
@neggert @adefazio