From e31e1ecf5759b43a2ee4ee2091628b940940965a Mon Sep 17 00:00:00 2001 From: 0x70b1a5 <0x70b1a5@users.noreply.github.com> Date: Mon, 21 Jan 2019 15:20:53 -0500 Subject: [PATCH] Add CLI argument for N_SECS --- train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/train.py b/train.py index 4483c10..2b85e4a 100644 --- a/train.py +++ b/train.py @@ -68,6 +68,7 @@ def get_arguments(): parser.add_argument('--emb_size', type=int, required=True) parser.add_argument('--rnn_type', choices=['LSTM', 'GRU'], required=True) parser.add_argument('--max_checkpoints', type=int, default=MAX_TO_KEEP) + parser.add_argument('--secs', type=int, default=N_SECS) return parser.parse_args()