INNER CODE UNIT · Python

main

lbcb-sci/RiNALMo · train_expression_level.py:135

def main(args):
    if args.seed:
        pl.seed_everything(args.seed)

    if args.output_dir:
        Path(args.output_dir).mkdir(parents=True, exist_ok=True)

    # Model
    model = ExpressionLevelWrapper(
        lm_config=args.lm_config,
        head_embed_dim=args.embed_dim,
        head_num_blocks=args.num_blocks,
        lr=args.lr,
    )

    if args.pretrained_rinalmo_weights:
        model.load_pretrained_lm_weights(args.pretrained_rinalmo_weights)

View source record →