pytorch resnet 全连接层linear报错：RuntimeError: mat1 dim 1 must match mat2 dim 0

    Traceback (most recent call last):
      File "/home/user1/pjs/frvt_pytorch/batch_run/2branch_alter_1update_2pfc_MMD_ori_auto/recognition/arcface_torch/tools/visualize.py", line 276, in <module>
    mask = grad_cam(input, target_index)
      File "/home/user1/pjs/frvt_pytorch/batch_run/2branch_alter_1update_2pfc_MMD_ori_auto/recognition/arcface_torch/tools/visualize.py", line 104, in __call__
    features, output = self.extractor(input.cuda())
      File "/home/user1/pjs/frvt_pytorch/batch_run/2branch_alter_1update_2pfc_MMD_ori_auto/recognition/arcface_torch/tools/visualize.py", line 6