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_ind