Pytorch学习系列之四 :卷积神经网络实现Minist数据集手写数值识别(导出为ONNX模型)

=== 全部的逻辑及解释都在代码及其注释之中,请仔细看注释!!!===



import torch as t
from torch.utils.data import DataLoader
import torchvision as tv
import cv2 as cv
import numpy as np
import os

'''$ Notice: torchvision 数据集的输出是范围在[0,1]之间的 PILImage,我们将他们转换成归一化范围为[-1,1]之间的张量 Tensors。
   $ transforms.

你可能感兴趣的:(Pytorch学习)