Skip to content
Snippets Groups Projects
Commit c7040c8c authored by Maria Guaranda-Cabezas's avatar Maria Guaranda-Cabezas
Browse files

adds corrections of params and summary writer

parent 2cc0b801
No related branches found
No related tags found
1 merge request!3Inference
......@@ -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)
......
......@@ -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,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment