diff --git a/script/get_train_list.py b/script/get_train_list.py
index 6e5872d..c40108d 100644
--- a/script/get_train_list.py
+++ b/script/get_train_list.py
@@ -26,4 +26,5 @@ def gen_train_file(args):
     parser.add_argument('--label_path', nargs='?', type=str, default=None)
     parser.add_argument('--img_path', nargs='?', type=str, default=None)
     parser.add_argument('--save_path', nargs='?', type=str, default=None)
-    args = parser.parse_args()
\ No newline at end of file
+    args = parser.parse_args()
+    gen_train_file(args)