Mercurial > repos > goeckslab > multimodal_learner

comparison multimodal_learner.py @ 6:871957823d0c 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 6e49ad44dd8572382ee203926690a30d7e888203
author goeckslab
date Mon, 26 Jan 2026 18:44:07 +0000
parents 975512caae22
children
comparison
equal deleted inserted replaced
5:975512caae22 6:871957823d0c
61 parser.add_argument("--cross_validation", type=str, default="false") 61 parser.add_argument("--cross_validation", type=str, default="false")
62 parser.add_argument("--num_folds", type=int, default=5) 62 parser.add_argument("--num_folds", type=int, default=5)
63 parser.add_argument("--epochs", type=int, default=None) 63 parser.add_argument("--epochs", type=int, default=None)
64 parser.add_argument("--learning_rate", type=float, default=None) 64 parser.add_argument("--learning_rate", type=float, default=None)
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,
67 help="DataLoader worker count (0 disables multiprocessing).")
68 parser.add_argument("--num_workers_eval", type=int, default=None,
69 help="DataLoader workers for evaluation; defaults to --num_workers.")
66 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")
67 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")
68 parser.add_argument("--validation_size", type=float, default=0.2) 72 parser.add_argument("--validation_size", type=float, default=0.2)
69 parser.add_argument("--split_probabilities", type=float, nargs=3, 73 parser.add_argument("--split_probabilities", type=float, nargs=3,
70 default=[0.7, 0.1, 0.2], metavar=("train", "val", "test")) 74 default=[0.7, 0.1, 0.2], metavar=("train", "val", "test"))
367 time_limit=args.time_limit, 371 time_limit=args.time_limit,
368 random_seed=args.random_seed, 372 random_seed=args.random_seed,
369 epochs=args.epochs, 373 epochs=args.epochs,
370 learning_rate=args.learning_rate, 374 learning_rate=args.learning_rate,
371 batch_size=args.batch_size, 375 batch_size=args.batch_size,
376 num_workers=args.num_workers,
377 num_workers_evaluation=args.num_workers_eval,
372 backbone_image=args.backbone_image, 378 backbone_image=args.backbone_image,
373 backbone_text=args.backbone_text, 379 backbone_text=args.backbone_text,
374 preset=args.preset, 380 preset=args.preset,
375 eval_metric=args.eval_metric, 381 eval_metric=args.eval_metric,
376 hyperparameters=args.hyperparameters, 382 hyperparameters=args.hyperparameters,