pytorch torchsummary 报错argument 0 is not a Variable,: Input type (torch.cuda.FloatTensor) and weigh

NoSuchKey