Yeonchan Ahn
commited on
Commit
•
af1ba97
1
Parent(s):
d40e838
fix error
Browse files
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 = (
|
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),
|