修改训练参数类型

2024-07-12 15:24:36 +08:00 · 2024-07-12 15:24:36 +08:00 · 9347fc29c0
parent 8c44c56cb6
commit 9347fc29c0
1 changed files with 1 additions and 1 deletions
--- a/msunet/app.py
+++ b/msunet/app.py
@ -36,7 +36,7 @@ class PredictOptions(BaseModel):

 class TrainOptions(BaseModel):
    train_path: InputFilePath = Field(..., ftypes=['zip'], description="训练数据集")
-    test_path: Optional[InputFilePath] = Field(ftypes=['ckpt'], description="测试使用的模型")
+    test_path: Optional[InputFilePath] = Field(ftypes=['zip'], description="测试使用的模型")
    output_dir: OutputDirectory = Field(
        default="./result"
    )