Pytorch框架卷积MNIST

Pytorch框架,卷积MNIST,使用CUDA,实测在这个计算量下,使用CUDA速度反而不如使用CPU,实测2轮测试,能到99%正确率左右 下面上程序 main.py #主程序 import numpy as np import random import os import torch import network from PIL import Image from torch.autograd import Variable import torch.nn as nn import time def img2vec(fname):#将图片转换成(0,1)的矩阵 im = Ima