Method get_epoch

custom_writer.py:70–71

(self)

        self.mode = mode

    def get_epoch(self) -> int:
        return self.epoch
    
    def get_keys(self, epoch: int = None) -> Tuple[str, ...]:
        """Returns keys1 e.g. train,eval."""