I have questions about adjusting the model and the loss function.

[lcxsnow]

I want to do some experiences to this project, like changing the model architecture and the loss function.
What do I suppose to add or delete or change the code, and what do I should be aware of?