For simplicity, ResNet18 from torchvision
is
used here. In real life applications, the process of creating a model can be quite
complicated.
from torchvision.models.resnet import resnet18
model = resnet18(pretrained=True)
For simplicity, ResNet18 from torchvision
is
used here. In real life applications, the process of creating a model can be quite
complicated.
from torchvision.models.resnet import resnet18
model = resnet18(pretrained=True)