upload files

2023-02-15 00:12:18 +08:00
parent 7dd4ce16c6
commit 823158e47b
3 changed files with 5 additions and 3 deletions
@@ -152,6 +152,8 @@ 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",
+                        help='finetune steps')

    args = parser.parse_args()
    model_dir = os.path.join("./", args.model)