MichaelDee

great. I looked at their code before, the model part should be straightforward, but they defined two their own loss functions, not sure where are the best places to adjust...