text_plain_output_1.png
import torch

import torchvision.models as models

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

cnn = models.vgg19(pretrained=True).features.to(device).eval()