From 72e1e8c44aed6f46db7143398e8b133597f1c939 Mon Sep 17 00:00:00 2001 From: Denny Britz Date: Sun, 25 Oct 2015 21:01:31 +0100 Subject: [PATCH] Remove unused var --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index eb8e413..af58596 100755 --- 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)