【torchsummary报错】RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.F
源代码:device=torch.device("cuda:0"iftorch.cuda.is_available()else"cpu")model=torchvision.models.resnet18(pretrained=None)model.fc=nn.Linear(512,10)summary(model,input_size=[(3,224,224)],batch_size=256,d