Merge pull request #10 from gokyori/master

supercoderhawk · web-flow · commit 4fb638b7cce9 · 2022-02-07T15:02:26.000+08:00
updated with missing argument for backend
diff --git a/deep_keyphrase/copy_rnn/train.py b/deep_keyphrase/copy_rnn/train.py
@@ -187,6 +187,7 @@ def parse_args(self, args=None):
         parser.add_argument('-prefetch', action='store_true', help='')
         parser.add_argument('-lazy_loading', action='store_true', help='')
         parser.add_argument('-fix_batch_size', action='store_true', help='')
+        parser.add_argument('-backend', type=str, default='tf', help='')
 
         # model specific parameter
         parser.add_argument("-embed_size", type=int, default=200, help='')
