I am trying to use the first few layers of trained resnet18, can anyone please help me with this?
I got an error while using .features, I assume it may have depreciated is there any other way to do so?
And also can anyone please help why the following error is showing up
image972×267 16.3 KB
import torch
from torchvision import transforms, models
from torchsummary import summary

print(summary(models.resnet18(pretrained=True),(3,224,224)))