Note
Go to the end to download the full example code.
Define Custom Model#
Defining your own model using Pytorch is very convenient. If you are not sure how to do it, you can refer to Dive into Deep Learning. Here, we will only introduce how to define the model so that it can be correctly used with trainers and experiments, as well as some useful tools.
TODO