Mercurial > repos > goeckslab > image_learner

diff image_learner_cli.py @ 21:d5c582cf74bc draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/gleam.git commit eed8c1e1d99a8a0c8f3a6bfdd8af48a5bfa19444
author goeckslab
date Tue, 20 Jan 2026 01:25:35 +0000
parents 64872c48a21f
children
line wrap: on
line diff
--- a/image_learner_cli.py	Tue Jan 06 15:35:11 2026 +0000
+++ b/image_learner_cli.py	Tue Jan 20 01:25:35 2026 +0000
@@ -172,6 +172,15 @@
             "to prevent data leakage (e.g., patient_id or slide_id)."
         ),
     )
+    parser.add_argument(
+        "--torch-home",
+        type=Path,
+        default=None,
+        help=(
+            "Directory for Torch Hub cache (pretrained weights). "
+            "Overrides TORCH_HOME for this run."
+        ),
+    )
 
     args = parser.parse_args()