From ea6e6ee1040fa85f743ab50b699fbeb04d9a0522 Mon Sep 17 00:00:00 2001 From: jules Date: Tue, 20 Mar 2018 23:35:18 +0100 Subject: scripts --- trainer/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'trainer/__init__.py') 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): -- cgit v1.2.3-70-g09d2