Imports

First, import the required modules:

import torch
from torch import nn

Then define the model:

class Model(nn.Module):
    def __init__(self):
        super().__init__()