change the directory for storing the saved model.
@@ -155,7 +155,7 @@ if __name__ == '__main__':
flags.DEFINE_integer('nb_epochs', 6, 'Number of epochs to train model')
flags.DEFINE_integer('batch_size', 128, 'Size of training batches')
flags.DEFINE_float('learning_rate', 0.1, 'Learning rate for training')
- flags.DEFINE_string('train_dir', './model', 'Directory storing the saved model.')
+ flags.DEFINE_string('train_dir', '/tmp', 'Directory storing the saved model.')
flags.DEFINE_string('filename', 'mnist.ckpt', 'Filename to save model under.')
flags.DEFINE_boolean('load_model', True, 'True for load, False for not load [False]')
app.run()
