chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update train.py

Browse Source
This commit is contained in:
Jeroen Oudshoorn
2023-08-29 15:09:59 +02:00
committed by GitHub
parent 4bc5be5159
commit e32b580166
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pwnagotchi/ai/train.py
View File

@ -181,7 +181,7 @@ class AsyncTrainer(object):
if os.path.isfile(self._nn_path): if os.path.isfile(self._nn_path):
back = "%s.bak" % self._nn_path back = "%s.bak" % self._nn_path
os.replace(self._nn_path, back) 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) self._model.learn(total_timesteps=epochs_per_episode, callback=self.on_ai_training_step)
except Exception as e: except Exception as e:
logging.exception("[ai] error while training (%s)", e) logging.exception("[ai] error while training (%s)", e)