diff options
| author | jules on spawn <spawn@cortex> | 2018-05-16 04:38:31 +0200 |
|---|---|---|
| committer | jules on spawn <spawn@cortex> | 2018-05-16 04:38:31 +0200 |
| commit | 566bd785e607b41d37120a20f236740944f00e84 (patch) | |
| tree | 510dfbaae561305a0ffa740f32d796c1ff5bde53 /trainer | |
| parent | 4df51dc9b69831a2422543a330e593c53ab14019 (diff) | |
fix trainer
Diffstat (limited to 'trainer')
| -rw-r--r-- | trainer/plugins.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/trainer/plugins.py b/trainer/plugins.py index 132c33d..5db48ab 100644 --- a/trainer/plugins.py +++ b/trainer/plugins.py @@ -161,7 +161,10 @@ class GeneratorPlugin(Plugin): self.generate = PrimedGenerator(trainer.model.model, trainer.cuda) def epoch(self, epoch_index): - samples = self.generate(self.n_samples, self.sample_length, self.primer, self.primer_a, self.primer_b, self.recursive) \ + if self.primer == "": + samples = self.generate(self.n_samples, self.sample_length).cpu().float().numpy() + else: + samples = self.generate(self.n_samples, self.sample_length, self.primer, self.primer_a, self.primer_b, self.recursive) \ .cpu().float().numpy() for i in range(self.n_samples): write_wav( |