def log_training(self, loss, learning_rate, iteration):
    self.add_scalar('training/loss', loss, iteration)
    self.add_scalar('learning_rate', learning_rate, iteration)
def log_training(self, loss, learning_rate, iteration):
    self.add_scalar('training/loss', loss, iteration)
    self.add_scalar('learning_rate', learning_rate, iteration)