From e32b580166c6a90c4b0ae707b3dfe294839d21a9 Mon Sep 17 00:00:00 2001 From: Jeroen Oudshoorn Date: Tue, 29 Aug 2023 15:09:59 +0200 Subject: [PATCH] Update train.py --- pwnagotchi/ai/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pwnagotchi/ai/train.py b/pwnagotchi/ai/train.py index c007230f..d8986393 100644 --- a/pwnagotchi/ai/train.py +++ b/pwnagotchi/ai/train.py @@ -181,7 +181,7 @@ 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._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)