Updated return value of get_loss functions for correctness (#70)
Updated return value of get_loss functions for correctness (#70)

* Updated return value of get_loss functions for correctness

* Adressed Davids comment

artemsok/sockeye,KellenSunderland/sockeye,awslabs/sockeye,artemsok/sockeye,artemsok/sockeye,KellenSunderland/sockeye,awslabs/sockeye,awslabs/sockeye