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

keep old version as backup when saving files

Browse Source
This commit is contained in:
Sniffleupagus
2023-04-27 15:00:58 -07:00
parent 62aaf6e3b1
commit c90855840c
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pwnagotchi/ai/train.py
View File

@ -77,9 +77,11 @@ class Stats(object):
}) })
temp = "%s.tmp" % self.path temp = "%s.tmp" % self.path
back = "%s.bak" % self.path
with open(temp, 'wt') as fp: with open(temp, 'wt') as fp:
fp.write(data) fp.write(data)
os.replace(self.path, back)
os.replace(temp, self.path) os.replace(temp, self.path)
@ -113,7 +115,9 @@ class AsyncTrainer(object):
def _save_ai(self): def _save_ai(self):
logging.info("[ai] saving model to %s ..." % self._nn_path) logging.info("[ai] saving model to %s ..." % self._nn_path)
temp = "%s.tmp" % self._nn_path temp = "%s.tmp" % self._nn_path
back = "%s.bak" % self._nn_path
self._model.save(temp) self._model.save(temp)
os.replace(self._nn_path, back)
os.replace(temp, self._nn_path) os.replace(temp, self._nn_path)
def on_ai_step(self): def on_ai_step(self):