Catch hyperparameter incompatibility in checkpoints (#319)

ziw-liu · web-flow · commit 8e557aca14dc · 2025-10-03T10:44:13.000-07:00
diff --git a/viscy/cli.py b/viscy/cli.py
@@ -36,6 +36,13 @@ def add_arguments_to_parser(self, parser) -> None:
             }
         )
 
+    def _parse_ckpt_path(self) -> None:
+        try:
+            return super()._parse_ckpt_path()
+        except SystemExit:
+            # FIXME: https://github.com/Lightning-AI/pytorch-lightning/issues/21255
+            return None
+
 
 def _setup_environment() -> None:
     """Set log level and TF32 precision."""

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,13 @@ def add_arguments_to_parser(self, parser) -> None:`
`36`	`36`	`}`
`37`	`37`	`)`
`38`	`38`
	`39`	`+ def _parse_ckpt_path(self) -> None:`
	`40`	`+ try:`
	`41`	`+ return super()._parse_ckpt_path()`
	`42`	`+ except SystemExit:`
	`43`	`+ # FIXME: https://github.com/Lightning-AI/pytorch-lightning/issues/21255`
	`44`	`+ return None`
	`45`	`+`
`39`	`46`
`40`	`47`	`def _setup_environment() -> None:`
`41`	`48`	`"""Set log level and TF32 precision."""`