INNER CODE UNIT · Python
_eval_step
lbcb-sci/RiNALMo · train_expression_level.py:86
def _eval_step(self, batch, batch_idx, log_prefix):
return self._common_step(batch, batch_idx, log_prefix=log_prefix)
def _on_eval_epoch_start(self):
# Reset metric calculator
self.r2_metric.reset()
self.spearman_metric.reset()
def _on_eval_epoch_end(self, log_prefix: str):
# Log and reset metric calculator
if not self.trainer.sanity_checking:
self.log(f"{log_prefix}/r2", self.r2_metric.compute(), sync_dist=True)
self.r2_metric.reset()
self.log(f"{log_prefix}/spearman", self.spearman_metric.compute(), sync_dist=True)
self.spearman_metric.reset()
def training_step(self, batch, batch_idx):
return self._common_step(batch, batch_idx, log_prefix="train")