
利用PyTorch和LSTM进行MNIST手写数字识别的例子
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目使用PyTorch框架结合长短时记忆网络(LSTM)模型,实现对手写数字图像的分类任务。通过训练,模型能够准确地从MNIST数据集中识别出0-9的手写数字。
代码如下:对于新手来说最重要的是学会RNN读取数据的格式。
# -*- coding: utf-8 -*-
Created on Tue Oct 9 08:53:25 2018
import sys
sys.path.append(..)
import torch
import datetime
from torch.autograd import Variable
from torch import nn
from torch.utils.data import DataLoader
from torchvision import transforms
全部评论 (0)
还没有任何评论哟~


