torchsummary : RuntimeError: Expected 5-dimensional input for 5-dimensional weight but...
torchsummary:RuntimeError:Expected5-dimensionalinputfor5-dimensionalweight[24,1,3,3,3],butgotinputofsize[2,1,1,16,256,256]instead.Solve:device=torch.device(“cuda”)#necessaryfortorchsummary,musttocudas