From 29da380ec204c39275beccb0a8c8c32a238b1faa Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Thu, 11 May 2017 11:47:28 +1000 Subject: [PATCH] update --- tutorial-contents/403_RNN_regressor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorial-contents/403_RNN_regressor.py b/tutorial-contents/403_RNN_regressor.py index f64e002..ad79500 100644 --- a/tutorial-contents/403_RNN_regressor.py +++ b/tutorial-contents/403_RNN_regressor.py @@ -16,7 +16,7 @@ import matplotlib.pyplot as plt torch.manual_seed(1) # reproducible # Hyper Parameters -TIME_STEP = 5 # rnn time step +TIME_STEP = 10 # rnn time step INPUT_SIZE = 1 # rnn input size LR = 0.02 # learning rate @@ -69,7 +69,7 @@ plt.show() for step in range(60): start, end = step * np.pi, (step+1)*np.pi # time steps # use sin predicts cos - steps = np.linspace(start, end, 10, dtype=np.float32) + steps = np.linspace(start, end, TIME_STEP, dtype=np.float32) x_np = np.sin(steps) # float32 for converting torch FloatTensor y_np = np.cos(steps)