sreddy-es

I got the same error for the following model: ``` class NeuralNetwork(nn.Module): def __init__(self): super(NeuralNetwork, self).__init__() self.flatten = nn.Flatten() self.linear_relu_stack = torch.nn.ModuleList([ torch.nn.Conv2d(3, 32, 3,stride=1, padding=1), torch.nn.ReLU(inplace=True), torch.nn.BatchNorm2d(32), torch.nn.Conv2d(32, 32,...