The model code is as follows:
def get_model():
    model = nn.Sequential(
            nn.Conv2d(3, 32, kernel_size=3, stride=2, padding=1),
            nn.ReLU(),
            nn.Conv2d(32, 64, kernel_size=3, stride=2, padding=1),
            nn.ReLU(),
            nn.Conv2d(64, 21, kernel_size=3, stride=2, padding=1),
            nn.ReLU(),
            nn.MaxPool2d(1)
    )
    opt = optim.SGD(model.parameters(), lr=0.1, momentum=0.9)
    return model, opt

It’s just a placeholder model for now, so it probably won’t yield very good results. The task is indeed image classification, for trying to identify 21 different types of land use.