PyTorch Forums
RuntimeError: Expected target size [6, 1, 224, 224], got [6, 3, 224, 224]
torch.package / torch::deploy
Tamim_Adnan
(Tamim Adnan)
January 17, 2023, 7:55am
1
RuntimeError: Expected target size [6, 1, 224, 224], got [6, 3, 224, 224]