From f2f606da0e04cc6c48246d34a0760fb18990f577 Mon Sep 17 00:00:00 2001
From: Harald Scheidl <harald_scheidl@hotmail.com>
Date: Fri, 15 Mar 2019 15:45:32 +0100
Subject: [PATCH] unify code style
---
.gitignore | 1 +
src/Model.py | 2 +-
src/main.py | 8 ++++----
3 files changed, 6 insertions(+), 5 deletions(-)
diff --git a/.gitignore b/.gitignore
index 6aedb48..7db9a63 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 eb3274a..c56e65f 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 0cec897..a33e255 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
--
GitLab