andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Morvan Zhou
2017-05-11 12:06:47 +10:00
committed by Morvan Zhou
parent 29da380ec2
commit 2523e0c1b1
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tutorial-contents/501_why_torch_dynamic_graph.py
View File

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