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),
|