Yeonchan Ahn commited on
Commit
af1ba97
1 Parent(s): d40e838
Files changed (1) hide show
  1. Alignment-and-Uniformity.py +1 -1
Alignment-and-Uniformity.py CHANGED
@@ -93,7 +93,7 @@ class AlignmentandUniformity(evaluate.Metric):
93
 
94
  nn_x_unif_loss_v = nonneg_uniform_loss(xs, t=self.unif_t)
95
  nn_y_unif_loss_v = nonneg_uniform_loss(ys, t=self.unif_t)
96
- nn_unif_loss = (nonneg_uniform_loss + nonneg_uniform_loss) / 2
97
 
98
  return {
99
  "align_loss": float(align_loss_val),
 
93
 
94
  nn_x_unif_loss_v = nonneg_uniform_loss(xs, t=self.unif_t)
95
  nn_y_unif_loss_v = nonneg_uniform_loss(ys, t=self.unif_t)
96
+ nn_unif_loss = (nn_x_unif_loss_v + nn_y_unif_loss_v) / 2
97
 
98
  return {
99
  "align_loss": float(align_loss_val),