Pytorch入门实战-----双向RNN识别手写数据集

窗口看到月雅湖

于 2019-06-25 10:46:37 发布

阅读量2k

点赞数 2

分类专栏： Pytroch实战

本文链接：https://blog.youkuaiyun.com/rongpeisheng666/article/details/93599944

版权

跑了一个epoch，测试了一下：

import torch
import torch.nn as nn
import torchvision.datasets as dsets
import torchvision.transforms as transforms
from torch.autograd import Variable

#Hyper Parameters
sequence_length = 28
input_size = 28
hidden_size = 128
num_layers = 2
num_classes = 10
batch_size = 100
num_epochs = 1
learning_rate = 0.003

#MNIST Datasets
train_dataset = dsets.MNIST(
    root='./data',
    train=True,
    transform=transforms.ToTensor(),
    download=False
)

test_dataset = dsets.MNIST(
    root='./data/',
    train=False,
    transform=transforms.ToTensor(),
    download=False
)

#Data Loader
train_loader = torch.utils.data.DataLoader(
    dataset= train_dataset,
    batch_size&