adds corrections of params and summary writer

src/diffwave/learner.py

@@ -148,7 +148,7 @@ class DiffWaveLearner:
   def _write_summary(self, step, features, loss):
     writer = self.summary_writer or SummaryWriter(self.model_dir, purge_step=step)
     # the following line will print a warning if the audio amplitude is out of range
-    writer.add_audio('feature/audio', features['audio'][0], step, sample_rate=self.params.sample_rate)
+    # writer.add_audio('feature/audio', features['audio'][0], step, sample_rate=self.params.sample_rate)
     if not self.params.unconditional:
       writer.add_image('feature/spectrogram', torch.flip(features['spectrogram'][:1], [1]), step)
     writer.add_scalar('train/loss', loss, step)

src/diffwave/params.py

@@ -34,7 +34,7 @@ class AttrDict(dict):
 
 params = AttrDict(
     # Training params
-    batch_size=16,
+    batch_size=64,
     learning_rate=2e-4,
     max_grad_norm=None,