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

No more Monitor,

Browse Source 
env hparams pickeling,
pomdp,
now training and learning
This commit is contained in:
steffen-illium
2021-06-01 16:38:55 +02:00
parent 55b409c72f
commit ff9846eb54
6 changed files with 95 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
reload_agent.py
View File

@@ -1,3 +1,4 @@
import pickle
import warnings
from pathlib import Path
@@ -12,10 +13,15 @@ warnings.filterwarnings('ignore', category=UserWarning)
if __name__ == '__main__':
dirt_props = DirtProperties()
env = SimpleFactory(n_agents=1, dirt_properties=dirt_props)
out_path = Path(r'C:\Users\steff\projects\f_iks\debug_out\PPO_1622485791\1_PPO_1622485791')
out_path = Path(r'C:\Users\steff\projects\f_iks\debug_out\A2C_1622557712')
with (out_path / f'env_{out_path.name}.pick').open('rb') as f:
env_kwargs = pickle.load(f)
env = SimpleFactory(**env_kwargs)
# Edit THIS:
model_path = out_path
model_files = list(natsorted(out_path.rglob('*.zip')))
this_model = model_files[0]