steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-09-13 22:44:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

pomdp

Browse Source
This commit is contained in:
steffen-illium
2021-06-01 12:39:33 +02:00
parent 403d38dc24
commit 55b409c72f
6 changed files with 82 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
reload_agent.py
View File

@@ -1,16 +1,11 @@
import warnings
from pathlib import Path
import time
from natsort import natsorted
from stable_baselines3 import PPO
from stable_baselines3.common.base_class import BaseAlgorithm
from stable_baselines3.common.callbacks import CallbackList
from stable_baselines3.common.evaluation import evaluate_policy
from environments.factory.simple_factory import DirtProperties, SimpleFactory
from environments.logging.monitor import MonitorCallback
from environments.logging.training import TraningMonitor
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@@ -20,7 +15,7 @@ if __name__ == '__main__':
dirt_props = DirtProperties()
env = SimpleFactory(n_agents=1, dirt_properties=dirt_props)
out_path = Path('debug_out')
out_path = Path(r'C:\Users\steff\projects\f_iks\debug_out\PPO_1622485791\1_PPO_1622485791')
model_files = list(natsorted(out_path.rglob('*.zip')))
this_model = model_files[0]