diff --git a/fig/mnist_acc.png b/fig/mnist_acc.png new file mode 100644 index 0000000..f308c29 Binary files /dev/null and b/fig/mnist_acc.png differ diff --git a/fig/mnist_ce.png b/fig/mnist_ce.png new file mode 100644 index 0000000..b8ccf07 Binary files /dev/null and b/fig/mnist_ce.png differ diff --git a/mnist_results.py b/mnist_results_basic.py similarity index 95% rename from mnist_results.py rename to mnist_results_basic.py index 83e88e8..13dcdf4 100644 --- a/mnist_results.py +++ b/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__':