diff --git a/rave/model.py b/rave/model.py index fbe66d5..2f4f4a3 100644 --- a/rave/model.py +++ b/rave/model.py @@ -317,6 +317,10 @@ def validation_step(self, batch, batch_idx): z = self.encoder.reparametrize(z)[0] y = self.decoder(z) + if self.valid_signal_crop and self.receptive_field.sum(): + x = rave.core.valid_signal_crop(x, *self.receptive_field) + y = rave.core.valid_signal_crop(y, *self.receptive_field) + if self.pqmf is not None: x = self.pqmf.inverse(x) y = self.pqmf.inverse(y)