fix kbits grid

facebookresearch · Jun 4, 2024 · 4890bc2 · 4890bc2
1 parent 0bc6808
commit 4890bc2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/audiocraft/grids/watermarking/_explorers.py b/audiocraft/grids/watermarking/_explorers.py
@@ -50,10 +50,10 @@ def get_grid_metrics(self):
             tt.group(
                 "evaluate",
                 [
-                    tt.leaf("acc", ".4f"),
-                    tt.leaf("fnr", ".4f"),
-                    tt.leaf("fpr", ".4f"),
-                    tt.leaf("bit_acc", ".4f"),
+                    tt.leaf("aug_identity_acc", ".4f"),
+                    tt.leaf("aug_identity_fnr", ".4f"),
+                    tt.leaf("aug_identity_fpr", ".4f"),
+                    tt.leaf("aug_identity_bit_acc", ".4f"),
                     tt.leaf("pesq", ".4f"),
                     tt.leaf("all_aug_acc", ".4f"),
                     tt.leaf("localization_acc_padding", ".4f"),