def _make_state_averager(self, **kwargs) -> TrainingStateAverager:
    return TrainingStateAverager(dht=self.dht, prefix=f'{self.run_id}_state_averager', min_matchmaking_time=self.matchmaking_time, allreduce_timeout=self.allreduce_timeout, shutdown_timeout=self.shutdown_timeout, offload_optimizer=self.offload_optimizer, custom_gradients=self.offload_optimizer, status_loglevel=self.status_loglevel, next_chunk_timeout=self.next_chunk_timeout, client_mode=self.client_mode, auxiliary=self.auxiliary, start=True, **kwargs)
def _make_state_averager(self, **kwargs) -> TrainingStateAverager:
    return TrainingStateAverager(dht=self.dht, prefix=f'{self.run_id}_state_averager', min_matchmaking_time=self.matchmaking_time, allreduce_timeout=self.allreduce_timeout, shutdown_timeout=self.shutdown_timeout, offload_optimizer=self.offload_optimizer, custom_gradients=self.offload_optimizer, status_loglevel=self.status_loglevel, next_chunk_timeout=self.next_chunk_timeout, client_mode=self.client_mode, auxiliary=self.auxiliary, start=True, **kwargs)