INNER CODE UNIT · Python
on_test_epoch_start
lbcb-sci/RiNALMo · train_expression_level.py:117
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,
"lr_scheduler": {
"scheduler": scheduler,
"interval": "step",
}
}