训练问题

[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