246152b8d459bb331f904b330e57311d3bf54672,cube/models/vocoder.py,BeeCoder,learn,#BeeCoder#,98

Before Change

            // fft_pred = torch.rfft(y_pred, 1)
            loss = self._get_loss(batch_y, y_pred, len(x))
            loss.backward()
            self.trainer.step()
            total_loss += loss
            num_batches += 1

        total_loss = total_loss.item()

After Change

            // fft_orig = torch.rfft(batch_y, 1)
            // fft_pred = torch.rfft(y_pred, 1)
            loss = self._get_loss(batch_y, y_pred, len(x))
            if loss is not None:
                loss.backward()
                self.trainer.step()
                total_loss += loss
                num_batches += 1

        total_loss = total_loss.item()
        return total_loss / num_batches

Instances