mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-07-05 09:01:36 +02:00
Adapt base_ac.py and utils.py to be compatible with refactored environment
This commit is contained in:
15
studies/marl_adapted.py
Normal file
15
studies/marl_adapted.py
Normal file
@ -0,0 +1,15 @@
|
||||
from pathlib import Path
|
||||
|
||||
from marl_factory_grid.algorithms.marl.iac import LoopIAC
|
||||
from marl_factory_grid.algorithms.utils import load_yaml_file
|
||||
|
||||
if __name__ == '__main__':
|
||||
cfg_path = Path('../marl_factory_grid/algorithms/marl/example_config.yaml')
|
||||
|
||||
cfg = load_yaml_file(cfg_path)
|
||||
|
||||
print("Training phase")
|
||||
agent = LoopIAC(cfg)
|
||||
agent.train_loop()
|
||||
print("Evaluation phase")
|
||||
agent.eval_loop(10)
|
Reference in New Issue
Block a user