INNER CODE UNIT · Python

on_validation_epoch_end

lbcb-sci/RiNALMo · train_ncrna_classification.py:125

    def on_validation_epoch_end(self):
        self._on_epoch_end(split='val')
    
    def test_step(self, batch, batch_idx):
        return self._validation_step(batch, batch_idx, split='test')
    
    def on_test_epoch_end(self):
        self._on_epoch_end(split='test')

    def configure_optimizers(self):
        optimizer = AdamW(
            filter(lambda p: p.requires_grad, self.parameters()),
            lr=self.lr,
            weight_decay=self.weight_decay
        )
        lr_scheduler = LinearLR(optimizer, start_factor=1.0, end_factor=1.0, total_iters=7500)

        return {

View source record →