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

v2.4.7

Browse Source 
Signed-off-by: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
This commit is contained in:
Jeroen Oudshoorn
2023-10-07 00:42:01 +02:00
parent eae6917c08
commit 74a6fa85e1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pwnagotchi/ai/__init__.py
View File

@ -21,12 +21,12 @@ def load(config, agent, epoch, from_disk=True):
SB_BACKEND = "stable_baselines3" SB_BACKEND = "stable_baselines3"
try: try:
from baselines.common.vec_env import DummyVecEnv from stable_baselines3.common.vec_env import DummyVecEnv
from baselines.a2c import a2c from stable_baselines3.a2c import a2c
logging.debug("[ai] A2C imported in %.2fs" % (time.time() - start)) logging.debug("[ai] A2C imported in %.2fs" % (time.time() - start))
start = time.time() start = time.time()
from baselines.a2c.policies import MlpPolicy from stable_baselines3.a2c.policies import MlpPolicy
logging.debug("[ai] MlpPolicy imported in %.2fs" % (time.time() - start)) logging.debug("[ai] MlpPolicy imported in %.2fs" % (time.time() - start))
SB_A2C_POLICY = MlpPolicy SB_A2C_POLICY = MlpPolicy