Spaces:
Runtime error
Runtime error
Commit
·
f6e5f6d
1
Parent(s):
236c7a5
Fix not logging FAD on test input data2
Browse files- remfx/models.py +2 -3
remfx/models.py
CHANGED
@@ -164,10 +164,9 @@ class RemFXModel(pl.LightningModule):
|
|
164 |
self.on_validation_batch_start(batch, batch_idx, dataloader_idx)
|
165 |
# Log FAD
|
166 |
x, target, label = batch
|
167 |
-
metric = self.metrics["FAD"]
|
168 |
self.log(
|
169 |
-
|
170 |
-
self.metrics[
|
171 |
on_step=False,
|
172 |
on_epoch=True,
|
173 |
logger=True,
|
|
|
164 |
self.on_validation_batch_start(batch, batch_idx, dataloader_idx)
|
165 |
# Log FAD
|
166 |
x, target, label = batch
|
|
|
167 |
self.log(
|
168 |
+
"Input_FAD",
|
169 |
+
self.metrics["FAD"](x, target),
|
170 |
on_step=False,
|
171 |
on_epoch=True,
|
172 |
logger=True,
|