This commit is contained in:
Morvan Zhou
2017-05-11 12:06:47 +10:00
committed by Morvan Zhou
parent 29da380ec2
commit 2523e0c1b1

View File

@ -16,11 +16,8 @@ import matplotlib.pyplot as plt
torch.manual_seed(1) # reproducible
# Hyper Parameters
BATCH_SIZE = 64
TIME_STEP = 5 # rnn time step / image height
INPUT_SIZE = 1 # rnn input size / image width
LR = 0.02 # learning rate
DOWNLOAD_MNIST = True # set to False if have downloaded the data
class RNN(nn.Module):