INNER CODE UNIT · Python
on_validation_epoch_end
lbcb-sci/RiNALMo · train_expression_level.py:111
def on_validation_epoch_end(self):
return self._on_eval_epoch_end("val")
def test_step(self, batch, batch_idx):
return self._eval_step(batch, batch_idx, log_prefix="test")
def on_test_epoch_start(self):
return self._on_eval_epoch_start()
def on_test_epoch_end(self):
return self._on_eval_epoch_end("test")
def configure_optimizers(self):
optimizer = Adam(filter(lambda p: p.requires_grad, self.parameters()), lr=self.lr)
scheduler = LinearLR(optimizer, start_factor=1.0, end_factor=0.1, total_iters=600) # TODO: Currently hardcoded!
return {
"optimizer": optimizer,