From 885495fcd14694ef6bcb6e4ff204b7ae8f486892 Mon Sep 17 00:00:00 2001 From: Jeroen Oudshoorn Date: Tue, 29 Aug 2023 14:19:18 +0200 Subject: [PATCH] Add files via upload --- pwnagotchi/ai/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pwnagotchi/ai/train.py b/pwnagotchi/ai/train.py index cba14dfe..c007230f 100644 --- a/pwnagotchi/ai/train.py +++ b/pwnagotchi/ai/train.py @@ -181,9 +181,8 @@ class AsyncTrainer(object): if os.path.isfile(self._nn_path): back = "%s.bak" % self._nn_path os.replace(self._nn_path, back) - self._view.set("mode", " ai") - self._model.learn(total_timesteps=epochs_per_episode, callback=self.on_ai_training_step) self._view.set("mode", " AI") + self._model.learn(total_timesteps=epochs_per_episode, callback=self.on_ai_training_step) except Exception as e: logging.exception("[ai] error while training (%s)", e) finally: