Пример #1
0
    def on_batch_start(self, state: State) -> None:
        """Batch start hook.

        Args:
            state (State): current state
        """
        state.batch_metrics = defaultdict(None)
Пример #2
0
    def on_batch_end(self, state: State) -> None:
        """Batch end hook.

        Args:
            state (State): current state
        """
        state.batch_metrics = self._process_metrics(state.batch_metrics)
        for key, value in state.batch_metrics.items():
            self.meters[key].add(value)
Пример #3
0
 def on_batch_end(self, state: State):
     state.batch_metrics = self._process_metrics(state.batch_metrics)
     for key, value in state.batch_metrics.items():
         self.meters[key].add(value)
Пример #4
0
 def on_batch_start(self, state: State):
     state.batch_metrics = defaultdict(None)