Skip to content

Commit cd83702

Browse files
refine aneurysm code for export and inference
1 parent 6aee31a commit cd83702

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

examples/aneurysm/aneurysm.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,6 @@ def export(cfg: DictConfig):
340340
# initialize solver
341341
solver = ppsci.solver.Solver(
342342
model,
343-
output_dir=cfg.output_dir,
344343
pretrained_model_path=cfg.INFER.pretrained_model_path,
345344
)
346345
# export model
@@ -401,7 +400,9 @@ def main(cfg: DictConfig):
401400
elif cfg.mode == "infer":
402401
inference(cfg)
403402
else:
404-
raise ValueError(f"cfg.mode should in ['train', 'eval'], but got '{cfg.mode}'")
403+
raise ValueError(
404+
f"cfg.mode should in ['train', 'eval', 'export', 'infer'], but got '{cfg.mode}'"
405+
)
405406

406407

407408
if __name__ == "__main__":

0 commit comments

Comments
 (0)