使用PyTorch实现简单的卷积神经网络识别MNIST手写数字

Pytorch 版本 0.4.1,系统:Ubuntu 16.04,Python版本:3.6 最近开始入门PyTorch,用了几天的时间学习,写了一个简单的手写数字识别练练手,参考了一个github的代码 如今作一个小小的总结和记录git 导入模块及定义全局变量 程序使用到的模块有:github import torch from torch import nn, optim # 网络及优化器 im
相关文章
相关标签/搜索