For simplicity, ResNet18 from torchvision is used here.
from torchvision.models.resnet import resnet18
model = resnet18(pretrained=True)
For simplicity, ResNet18 from torchvision is used here.
from torchvision.models.resnet import resnet18
model = resnet18(pretrained=True)