Hi, I just used summary to output the information about my model, but it did not work. Then, I tested it with an official example, and it did not work too. For example,
from torchsummary import summary
model=torchvision.models.vgg16()
model=model.cuda()
summary(model,(3,224,224))
The error is “can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first”
In a word, the function “summary” seems does not work on my machine. I do not know why it happened. Please help me. Thanks!