scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Simone Margaritelli 2019-10-01 20:18:09 +02:00
parent efc1997f02
commit 8c840f9c82
1 changed files with 2 additions and 2 deletions

4
sdcard/rootfs/root/pwnagotchi/scripts/pwnagotchi/ai/train.py

@ -77,7 +77,7 @@ class Stats(object):
} }
}) })
temp = ".%s.tmp" % self.path temp = "%s.tmp" % self.path
with open(temp, 'wt') as fp: with open(temp, 'wt') as fp:
fp.write(data) fp.write(data)
@ -109,7 +109,7 @@ class AsyncTrainer(object):
def _save_ai(self): def _save_ai(self):
core.log("[ai] saving model to %s ..." % self._nn_path) core.log("[ai] saving model to %s ..." % self._nn_path)
temp = ".%s.tmp" % self._nn_path temp = "%s.tmp" % self._nn_path
self._model.save(temp) self._model.save(temp)
os.replace(temp, self._nn_path) os.replace(temp, self._nn_path)