From 6617d5828d046f3cfb6f4e0a2e9385086494662c Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Mon, 23 Aug 2021 08:30:39 +0800
Subject: [PATCH] Train more epochs for GitHub actions.

---
 egs/yesno/ASR/tdnn/train.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/egs/yesno/ASR/tdnn/train.py b/egs/yesno/ASR/tdnn/train.py
index 7cce8a54a..603085fcd 100755
--- a/egs/yesno/ASR/tdnn/train.py
+++ b/egs/yesno/ASR/tdnn/train.py
@@ -59,6 +59,13 @@ def get_parser():
         help="Should various information be logged in tensorboard.",
     )
 
+    parser.add_argument(
+        "--num-epochs",
+        type=int,
+        default=50,
+        help="Number of epochs to train.",
+    )
+
     return parser