upload files

2023-02-15 01:36:30 +08:00
parent 823158e47b
commit 1418b66b28
1 changed files with 2 additions and 1 deletions
@@ -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