Open rochaie opened 1 year ago
More details?
RuntimeError: The size of tensor a (2) must match the size of tensor b (16) at non-singleton dimension 3; After removes optimizer 's param weight_decay, the model run normally,but training abnormal How should I modify it?
Change the version of pytorch. The problem can be solved.
@xiaoyuxie-vico What is the PyTorch version required?
@xiaoyuxie-vico What is the PyTorch version required?
Pytorch1.13.0+cu116 works for me.
RuntimeError: The size of tensor a (2) must match the size of tensor b (16) at non-singleton dimension 3; After removes optimizer 's param weight_decay, the model run normally,but training abnormal How should I modify it?