diff options
| author | jules <jules@asdf.us> | 2019-08-22 16:31:48 +0200 |
|---|---|---|
| committer | jules <jules@asdf.us> | 2019-08-22 16:31:48 +0200 |
| commit | 3a956cbc3c08566c242a1fbaf786a73381d7239f (patch) | |
| tree | b571713a791afb24ed7f94b91ae4a415cc37d626 /trainer | |
| parent | df754e0c33fc594706bea44d936a58b7b1cd8f2f (diff) | |
| parent | eeed56591801c73b34f968cd98f1b48b01997050 (diff) | |
Merge branch 'master' of asdf.us:samplernn
Diffstat (limited to 'trainer')
| -rw-r--r-- | trainer/plugins.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/trainer/plugins.py b/trainer/plugins.py index 5db48ab..d4cac9e 100644 --- a/trainer/plugins.py +++ b/trainer/plugins.py @@ -161,9 +161,11 @@ class GeneratorPlugin(Plugin): self.generate = PrimedGenerator(trainer.model.model, trainer.cuda) def epoch(self, epoch_index): + print("Generating {} x {} samples".format(self.n_samples, self.sample_length)) if self.primer == "": samples = self.generate(self.n_samples, self.sample_length).cpu().float().numpy() else: + print("Using primer {}".format(self.primer)) 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): |