RuntimeError                              Traceback (most recent call last)
 in 
27 #         print(‘number of epoch’, epoch, ‘loss’, loss.item())
28
—> 29 predict_out = net(torch.Tensor(test_X))
30 _, predict_y = torch.max(predict_out, 1)
31
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torch/nn/modules/module.py in call(self, *input, **kwargs)
530             result = self._slow_forward(*input, **kwargs)
531         else:
 → 532             result = self.forward(*input, **kwargs)
533         for hook in self._forward_hooks.values():
534             hook_result = hook(self, input, result)
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torchvision/models/resnet.py in forward(self, x)
214
215     def forward(self, x):
 → 216         return self._forward_impl(x)
217
218
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torchvision/models/resnet.py in _forward_impl(self, x)
197     def _forward_impl(self, x):
198         # See note [TorchScript super()]
 → 199         x = self.conv1(x)
200         x = self.bn1(x)
201         x = self.relu(x)
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torch/nn/modules/module.py in call(self, *input, **kwargs)
530             result = self._slow_forward(*input, **kwargs)
531         else:
 → 532             result = self.forward(*input, **kwargs)
533         for hook in self._forward_hooks.values():
534             hook_result = hook(self, input, result)
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torch/nn/modules/conv.py in forward(self, input)
343
344     def forward(self, input):
 → 345         return self.conv2d_forward(input, self.weight)
346
347 class Conv3d(_ConvNd):
/scratch/adomakor412/conda/envs/MyEnv/lib/python3.6/site-packages/torch/nn/modules/conv.py in conv2d_forward(self, input, weight)
340                             _pair(0), self.dilation, self.groups)
341         return F.conv2d(input, weight, self.bias, self.stride,
 → 342                         self.padding, self.dilation, self.groups)
343
344     def forward(self, input):
RuntimeError: Expected 4-dimensional input for 4-dimensional weight 64 3 7 7, but got 1-dimensional input of size [1455] instead