INNER CODE UNIT · Python

save_top_k

lbcb-sci/RiNALMo · train_expression_level.py:189

            save_top_k=-1
        )
        callbacks.append(epoch_ckpt_callback)

    if args.checkpoint_every_epoch_top_1:
        epoch_ckpt_callback = ModelCheckpoint(
            dirpath=args.output_dir,
            filename='te-fold{args.fold}-top1',
            every_n_epochs=1,
            save_top_k=1,
            monitor='val/r2',
            mode='max'
        )
        callbacks.append(epoch_ckpt_callback)

    if loggers:
        lr_monitor = LearningRateMonitor(logging_interval="step")
        callbacks.append(lr_monitor)

View source record →