@property
def state(self):
    return dict(self.state_averager.optimizer.state, local_epoch=self.local_epoch)
@property
def state(self):
    return dict(self.state_averager.optimizer.state, local_epoch=self.local_epoch)