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

misc: small fix or general refactoring i did not bother commenting

Browse Source
This commit is contained in:
Simone Margaritelli 2019-10-25 12:17:44 +02:00
parent 7954bb5fcf
commit 8b00e0ae10
No known key found for this signature in database
GPG Key ID: 82E42E7F3B34C97E
1 changed files with 5 additions and 0 deletions

5
pwnagotchi/ai/__init__.py

@ -16,6 +16,8 @@ def load(config, agent, epoch, from_disk=True):
return False return False
try: try:
begin = time.time()
logging.info("[ai] bootstrapping dependencies ...") logging.info("[ai] bootstrapping dependencies ...")
start = time.time() start = time.time()
@ -53,8 +55,11 @@ def load(config, agent, epoch, from_disk=True):
for key, value in config['params'].items(): for key, value in config['params'].items():
logging.info(" %s: %s" % (key, value)) logging.info(" %s: %s" % (key, value))
logging.debug("[ai] total loading time is %.2fs" % (time.time() - begin))
return a2c return a2c
except Exception as e: except Exception as e:
logging.exception("error while starting AI") logging.exception("error while starting AI")
logging.warning("[ai] AI not loaded!")
return False return False