model_pyt = models.alexnet(pretrained=True).eval()
if torch.cuda.is_available():
model_pyt = model_pyt.cuda()
cuda加载model
于 2022-04-21 19:28:51 首次发布
model_pyt = models.alexnet(pretrained=True).eval()
if torch.cuda.is_available():
model_pyt = model_pyt.cuda()