steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-12 10:30:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactoring and init.py

Browse Source
This commit is contained in:
Steffen Illium
2023-06-20 18:21:43 +02:00
parent 1332cee7e1
commit c7d77acbbe
138 changed files with 328 additions and 320 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
reload_agent.py
View File

@@ -2,13 +2,12 @@ import warnings
from pathlib import Path
import yaml
from stable_baselines3 import PPO
from environment.factory import BaseFactory
from environment.logging.envmonitor import EnvMonitor
from environment.logging.recorder import EnvRecorder
from mfg_package.environment.factory import BaseFactory
from mfg_package.logging.envmonitor import EnvMonitor
from mfg_package.logging.recorder import EnvRecorder
from modules.doors import constants as d
from mfg_package.modules.doors import constants as d
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@@ -32,7 +31,7 @@ if __name__ == '__main__':
this_model = out_path / 'model.zip'
model_cls = PPO # next(val for key, val in h.MODEL_MAP.items() if key in out_path.parent.name)
model_cls = None # next(val for key, val in h.MODEL_MAP.items() if key in out_path.parent.name)
models = [model_cls.load(this_model)]
try:
# Legacy Cleanups