Loss v len callback (#1226)
* adding loss v len callback
* lint
* lint
* using model's loss function
* lint
* only log to mlflow logger
* incorporating suggestions
* incorporating comments
---------
Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com>
