import torch
from torchvision.models import resnet50