"""
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 <https://d2l.ai/>`_.
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