import cv2
import torch
import torchvision
from torchvision import datasets
from torchvision import transforms
from torch.autograd import Variable
import torch.nn as nn
transform = transforms.Compose([transforms.ToTensor(),
transforms.Normalize(mean=[0.5], std=[0.5])])
data_test = datasets.MNIST(root="./data/",
transform=transform,
train=False)
data_loader_test = torch.utils.data.DataLoader(dataset=data_test,
batch_size=64,
shuffle=True,
drop_last&#
导入训练好的模型,Pytorch
最新推荐文章于 2024-04-13 10:27:15 发布