mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-07-05 09:01:36 +02:00
15 lines
420 B
Python
15 lines
420 B
Python
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) |