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

move brain.nn backup into ai_worker before training session instead of backing up on every save. keep the "previous trained" brain

Browse Source
This commit is contained in:
Sniffleupagus
2023-07-04 13:18:18 -07:00
parent 9978d5da36
commit cd92491e40
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pwnagotchi/ai/train.py
View File

@ -116,10 +116,7 @@ class AsyncTrainer(object):
def _save_ai(self):
logging.info("[ai] saving model to %s ..." % self._nn_path)
temp = "%s.tmp" % self._nn_path
back = "%s.bak" % self._nn_path
self._model.save(temp)
if os.path.isfile(self._nn_path):
os.replace(self._nn_path, back)
os.replace(temp, self._nn_path)
def on_ai_step(self):
@ -180,7 +177,13 @@ class AsyncTrainer(object):
logging.info("[ai] learning for %d epochs ..." % epochs_per_episode)
try:
self.set_training(True, epochs_per_episode)
# back up brain file before starting new training set
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")
except Exception as e:
logging.exception("[ai] error while training (%s)", e)
finally: