From 566bd785e607b41d37120a20f236740944f00e84 Mon Sep 17 00:00:00 2001
From: jules on spawn <spawn@cortex>
Date: Wed, 16 May 2018 04:38:31 +0200
Subject: fix trainer

---
 trainer/plugins.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'trainer')

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(
-- 
cgit v1.2.3-70-g09d2