Can the loss function be changed? #65
Description
I noticed that the current loss function is passed to prepare_model() via Config.train_loss, and the default is set to MSE.
{{ message }}
I noticed that the current loss function is passed to prepare_model() via Config.train_loss, and the default is set to MSE.