... std=[0.229, 0.224, 0.225])]) img_tensor = transform(img) batch = img_tensor.unsqueeze(0) model = models.alexnet(pretrained=True) device = "cuda" if ...
確定! 回上一頁