class Model(torch.nn.Module): def __init__(self): super(Model, self).__init__() self.conv1 = torch.nn.Sequential(torch.nn.Conv2d(1, 4, kernel_size=5, stride=1,padding=0), torch.nn.MaxPool2d(stride=4, kernel_size=4)) self.dense = torch.nn.Sequential(t