scripts

1 files changed, 9 insertions, 0 deletions

diff --git a/trainer/__init__.py b/trainer/__init__.py
index 7e2ea18..1f39506 100644
--- a/trainer/__init__.py
+++ b/trainer/__init__.py
@@ -56,6 +56,15 @@ class Trainer(object):
             self.train()
             self.call_plugins('epoch', self.epochs)
 
+    def generate(self, epochs=1):
+        for q in self.plugin_queues.values():
+            heapq.heapify(q)
+
+        for self.epochs in range(self.epochs + 1, self.epochs + epochs + 1):
+            # self.train()
+            self.call_plugins('update', self.iterations, self.model)
+            self.call_plugins('epoch', self.epochs)
+
     def train(self):
         for (self.iterations, data) in \
                 enumerate(self.dataset, self.iterations + 1):