mattricesound commited on
Commit
f6e5f6d
·
1 Parent(s): 236c7a5

Fix not logging FAD on test input data2

Browse files
Files changed (1) hide show
  1. 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
- f"Input_{metric}",
170
- self.metrics[metric](x, target),
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,