Mercurial > repos > goeckslab > multimodal_learner

comparison 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
comparison
equal deleted inserted replaced
6:871957823d0c 7:ed2fefc8d892
65 parser.add_argument("--batch_size", type=int, default=None) 65 parser.add_argument("--batch_size", type=int, default=None)
66 parser.add_argument("--num_workers", type=int, default=None, 66 parser.add_argument("--num_workers", type=int, default=None,
67 help="DataLoader worker count (0 disables multiprocessing).") 67 help="DataLoader worker count (0 disables multiprocessing).")
68 parser.add_argument("--num_workers_eval", type=int, default=None, 68 parser.add_argument("--num_workers_eval", type=int, default=None,
69 help="DataLoader workers for evaluation; defaults to --num_workers.") 69 help="DataLoader workers for evaluation; defaults to --num_workers.")
70 parser.add_argument("--backbone_image", type=str, default="swin_base_patch4_window7_224") 70 parser.add_argument("--backbone_image", type=str, default="swin_base_patch4_window7_224.ms_in22k_ft_in1k")
71 parser.add_argument("--backbone_text", type=str, default="microsoft/deberta-v3-base") 71 parser.add_argument("--backbone_text", type=str, default="microsoft/deberta-v3-base")
72 parser.add_argument("--validation_size", type=float, default=0.2) 72 parser.add_argument("--validation_size", type=float, default=0.2)
73 parser.add_argument("--split_probabilities", type=float, nargs=3, 73 parser.add_argument("--split_probabilities", type=float, nargs=3,
74 default=[0.7, 0.1, 0.2], metavar=("train", "val", "test")) 74 default=[0.7, 0.1, 0.2], metavar=("train", "val", "test"))
75 parser.add_argument("--sample_id_column", default=None) 75 parser.add_argument("--sample_id_column", default=None)