src/proteingnn/model.py
def training_epoch_end(self, outputs: Dict[str, Any], **kwargs) -> None:
        """Aggregate training step logging information and plot."""
        if self.global_step == 0:
            self.dual_dict['train'] = None
            return

        if self.log_train:
            outputs = _concat_step_output(outputs)
            if self._regression:
                self._log_scaler_reg(outputs['y'], outputs['pred'], stage='train')
            else:
                self._log_scaler_cls(outputs['y'], outputs['pred'], stage='train', **kwargs)
                self._log_pr_curve(outputs['y'], outputs['pred'])

            self.dual_dict['train'] = outputs
Aggregate training step logging information and plot.
Aggregate training step logging information and plot.
Aggregate training step logging information and plot .
def training_epoch_end(self, outputs: Dict[str, Any], **kwargs) -> None:
        if self.global_step == 0:
            self.dual_dict['train'] = None
            return
        if self.log_train:
            outputs = _concat_step_output(outputs)
            if self._regression:
                self._log_scaler_reg(outputs['y'], outputs['pred'], stage='train')
            else:
                self._log_scaler_cls(outputs['y'], outputs['pred'], stage='train', **kwargs)
                self._log_pr_curve(outputs['y'], outputs['pred'])
            self.dual_dict['train'] = outputs
def training_epoch_end ( self , outputs : Dict [ str , Any ] , ** kwargs ) -> None : if self . global_step == 0 : self . dual_dict [ 'train' ] = None return if self . log_train : outputs = _concat_step_output ( outputs ) if self . _regression : self . _log_scaler_reg ( outputs [ 'y' ] , outputs [ 'pred' ] , stage = 'train' ) else : self . _log_scaler_cls ( outputs [ 'y' ] , outputs [ 'pred' ] , stage = 'train' , ** kwargs ) self . _log_pr_curve ( outputs [ 'y' ] , outputs [ 'pred' ] ) self . dual_dict [ 'train' ] = outputs
Aggregate training step logging information and plot.
Aggregate training step logging information and plot .
"""Aggregate training step logging information and plot."""
self
outputs
Dict[str, Any]