Remove unused var

vilhub · Oct 25, 2015 · 72e1e8c · 72e1e8c
1 parent 6ba5495
commit 72e1e8c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train.py b/train.py
@@ -9,7 +9,6 @@
 from gru_theano import GRUTheano
 
 LEARNING_RATE = float(os.environ.get("LEARNING_RATE", "0.001"))
-SUBSAMPLE = int(os.environ.get("SUBSAMPLE")) if os.environ.get("SUBSAMPLE") else None
 VOCABULARY_SIZE = int(os.environ.get("VOCABULARY_SIZE", "2000"))
 EMBEDDING_DIM = int(os.environ.get("EMBEDDING_DIM", "48"))
 HIDDEN_DIM = int(os.environ.get("HIDDEN_DIM", "128"))
@@ -48,6 +47,6 @@ def sgd_callback(model, num_examples_seen):
   sys.stdout.flush()
 
 for epoch in range(NEPOCH):
-  train_with_sgd(model, x_train[:5001], y_train[:5001], learning_rate=LEARNING_RATE, nepoch=1, decay=0.9, 
+  train_with_sgd(model, x_train, y_train, learning_rate=LEARNING_RATE, nepoch=1, decay=0.9, 
     callback_every=PRINT_EVERY, callback=sgd_callback)