Add Huber loss, allow choosing training loss function from the yaml

Add Huber loss, allow choosing training loss function from the yaml #336