训练问题

[aprillysea]

Traceback (most recent call last): File "E:\code\PFLD-pytorch-master\train.py", line 221, in main(args) File "E:\code\PFLD-pytorch-master\train.py", line 146, in main optimizer, epoch) File "E:\code\PFLD-pytorch-master\train.py", line 61, in train features, landmarks = pfld_backbone(img) File "C:\Users\user.conda\envs\landmark\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "E:\code\PFLD-pytorch-master\models\pfld.py", line 103, in forward x = self.relu(self.bn1(self.conv1(x))) # [64, 56, 56] File "C:\Users\user.conda\envs\landmark\lib\site-packages\torch\nn\modules\module.py", line 1194, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\user.conda\envs\landmark\lib\site-packages\torch\nn\modules\conv.py", line 463, in forward return self._conv_forward(input, self.weight, self.bias) File "C:\Users\user.conda\envs\landmark\lib\site-packages\torch\nn\modules\conv.py", line 460, in _conv_forward self.padding, self.dilation, self.groups) RuntimeError: Given groups=1, weight of size [64, 3, 3, 3], expected input[256, 112, 112, 3] to have 3 channels, but got 112 channels instead