MNIST

philipperemy · Dec 19, 2016 · 2e7bfe3 · 2e7bfe3
1 parent 86555aa
commit 2e7bfe3
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 2 deletions.
diff --git a/fig/mnist_acc.png b/fig/mnist_acc.png
diff --git a/fig/mnist_ce.png b/fig/mnist_ce.png
diff --git a/mnist_results.py → mnist_results_basic.py b/mnist_results.py → mnist_results_basic.py
@@ -6,7 +6,6 @@
 from basic_lstm import BasicLSTMCell
 from helpers import FileLogger
 from ml_utils import create_convolution_variable, create_bias_variable
-from phased_lstm import PhasedLSTMCell
 
 
 def run_lstm_mnist(lstm_cell=BasicLSTMCell, hidden_size=32, batch_size=256, steps=20):
@@ -57,7 +56,7 @@ def transform_x(_x_):
 
 def main():
     # Vanilla LSTM
-    run_lstm_mnist(lstm_cell=PhasedLSTMCell, hidden_size=32, batch_size=256, steps=2000)
+    run_lstm_mnist(lstm_cell=BasicLSTMCell, hidden_size=32, batch_size=256, steps=2000)
 
 
 if __name__ == '__main__':