typo

jhauret · Dec 19, 2024 · ddc7222 · ddc7222
1 parent 8bfa965
commit ddc7222
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vibravox/lightning_modules/eben.py b/vibravox/lightning_modules/eben.py
@@ -301,7 +301,7 @@ def common_eval_step(self, batch, batch_idx, stage):
         )
 
         for key, value in atomic_losses_discriminator.items():
-            self.log(f"train/discriminator/{key}", value, sync_dist=True)
+            self.log(f"{stage}/discriminator/{key}", value, sync_dist=True)
 
         return outputs