def write_loss(self, loss_name, scalar, n_iter):
    self.add_scalar(('Loss/' + loss_name), scalar, n_iter)
def write_loss(self, loss_name, scalar, n_iter):
    self.add_scalar(('Loss/' + loss_name), scalar, n_iter)