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

Item debugging and New Entities

Browse Source
This commit is contained in:
Steffen Illium
2021-08-23 18:39:50 +02:00
parent d5e4d44823
commit c3d4925653
4 changed files with 43 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
reload_agent.py
View File

@@ -7,6 +7,7 @@ from stable_baselines3 import PPO
from stable_baselines3.common.evaluation import evaluate_policy
from environments.factory.simple_factory import DirtProperties, SimpleFactory
from environments.factory.double_task_factory import ItemProperties, DoubleTaskFactory
warnings.filterwarnings('ignore', category=FutureWarning)
warnings.filterwarnings('ignore', category=UserWarning)
@@ -14,7 +15,7 @@ warnings.filterwarnings('ignore', category=UserWarning)
if __name__ == '__main__':
model_name = 'A2C_1627491061'
model_name = 'A2C_1629467677'
run_id = 0
out_path = Path(__file__).parent / 'debug_out'
model_path = out_path / model_name
@@ -26,7 +27,7 @@ if __name__ == '__main__':
max_local_amount=1, spawn_frequency=5, max_spawn_ratio=0.05,
dirt_smear_amount=0.5),
combin_agent_slices_in_obs=True, omit_agent_slice_in_obs=True)
with SimpleFactory(**env_kwargs) as env:
with DoubleTaskFactory(**env_kwargs) as env:
# Edit THIS:
model_files = list(natsorted((model_path / f'{run_id}_{model_name}').rglob('model_*.zip')))