diff --git a/.gitignore b/.gitignore index 6aedb482aaf032249e159194ad6a44b9b5f0dc0c..7db9a633d0fd93ef2b92d79ee536a6af262c9d49 100644 --- a/.gitignore +++ b/.gitignore @@ -6,3 +6,4 @@ model/snapshot-* notes/ *.so *.pyc +.idea/ \ No newline at end of file diff --git a/src/Model.py b/src/Model.py index eb3274af3327e50e1e3baa61af3d31e71b38f898..c56e65f46d60f0a55b461ef14547ca47198a5f46 100644 --- a/src/Model.py +++ b/src/Model.py @@ -28,7 +28,7 @@ class Model: self.snapID = 0 # Whether to use normalization over a batch or a population - self.is_train = tf.placeholder(tf.bool, name="is_train"); + self.is_train = tf.placeholder(tf.bool, name='is_train') # input image batch self.inputImgs = tf.placeholder(tf.float32, shape=(None, Model.imgSize[0], Model.imgSize[1])) diff --git a/src/main.py b/src/main.py index 0cec89782bc09e7b71826bdcc1a90c458201ebe7..a33e255bfc30f0dd81bbdaf5b53b1b30a49e3d1e 100644 --- a/src/main.py +++ b/src/main.py @@ -101,10 +101,10 @@ def main(): "main function" # optional command line args parser = argparse.ArgumentParser() - parser.add_argument("--train", help="train the NN", action="store_true") - parser.add_argument("--validate", help="validate the NN", action="store_true") - parser.add_argument("--beamsearch", help="use beam search instead of best path decoding", action="store_true") - parser.add_argument("--wordbeamsearch", help="use word beam search instead of best path decoding", action="store_true") + parser.add_argument('--train', help='train the NN', action='store_true') + parser.add_argument('--validate', help='validate the NN', action='store_true') + parser.add_argument('--beamsearch', help='use beam search instead of best path decoding', action='store_true') + parser.add_argument('--wordbeamsearch', help='use word beam search instead of best path decoding', action='store_true') args = parser.parse_args() decoderType = DecoderType.BestPath