image_output_1.png
from torchvision import models, transforms, datasets

import torch

torch.__version__

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

inputs_try.shape

model_vgg = models.vgg16(pretrained=True)

inputs_try, lables_try = (inputs_try.to(device), labels_try.to(device))
model_vgg = model_vgg.to(device)

outputs_try = model_vgg(inputs_try)

outputs_try