From 1418b66b28282c72b83dd728354b2e2a35d088e2 Mon Sep 17 00:00:00 2001
From: Plachta <liusongting07@gmail.com>
Date: Wed, 15 Feb 2023 01:36:30 +0800
Subject: [PATCH] upload files

---
 utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index c2a3113..4550728 100644
--- a/utils.py
+++ b/utils.py
@@ -152,7 +152,7 @@ def get_hparams(init=True):
                         help='JSON file for configuration')
     parser.add_argument('-m', '--model', type=str, default="pretrained_models",
                         help='Model name')
-    parser.add_argument('-n', '--n_steps', type=str, default="2000",
+    parser.add_argument('-n', '--n_steps', type=int, default="2000",
                         help='finetune steps')
 
     args = parser.parse_args()
@@ -175,6 +175,7 @@ def get_hparams(init=True):
 
     hparams = HParams(**config)
     hparams.model_dir = model_dir
+    hparams.n_steps = args.n_steps
     return hparams