Mercurial > repos > goeckslab > multimodal_learner
comparison multimodal_learner.py @ 7:ed2fefc8d892 draft default tip
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) |