Mercurial > repos > goeckslab > multimodal_learner

diff multimodal_learner.py @ 7:ed2fefc8d892 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 7e1fa6cd06c388e2592153f662187625797610e9
author goeckslab
date Wed, 28 Jan 2026 19:56:37 +0000
parents 871957823d0c
children
line wrap: on
line diff
--- a/multimodal_learner.py	Mon Jan 26 18:44:07 2026 +0000
+++ b/multimodal_learner.py	Wed Jan 28 19:56:37 2026 +0000
@@ -67,7 +67,7 @@
                         help="DataLoader worker count (0 disables multiprocessing).")
     parser.add_argument("--num_workers_eval", type=int, default=None,
                         help="DataLoader workers for evaluation; defaults to --num_workers.")
-    parser.add_argument("--backbone_image", type=str, default="swin_base_patch4_window7_224")
+    parser.add_argument("--backbone_image", type=str, default="swin_base_patch4_window7_224.ms_in22k_ft_in1k")
     parser.add_argument("--backbone_text", type=str, default="microsoft/deberta-v3-base")
     parser.add_argument("--validation_size", type=float, default=0.2)
     parser.add_argument("--split_probabilities", type=float, nargs=3,