import tensorflow as tf

        os.makedirs('./mnist')

    with tf.Session() as sess:
        if args.model_path == '':
            tf.global_variables_initializer().run()
        else:
            saver.restore(sess, args.model_path)

