-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
31 lines (26 loc) · 838 Bytes
/
main.py
File metadata and controls
31 lines (26 loc) · 838 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from src.environment import OptimizationEnv
from src.mappo import MAPPO
from src.deephive import DeepHive
from distutils.util import strtobool
import os
import argparse
import numpy as np
from commons.utils import get_config, get_args, get_environment, get_policy
def main(title, env, policy, mode, config, **kwargs):
deephive = DeepHive(title, env, policy, mode, config, **kwargs)
deephive.optimize(debug=True)
if __name__ == '__main__':
args = get_args()
title = args.title
env_name = args.env
mode = args.mode
reinit = args.reinit
log = args.log
tags = args.tags
config = get_config(args.config)
# ENVIRONMENT
env = get_environment(config, env_name, reinit=reinit)
# POLICY
policy = get_policy(config, mode)
# MAIN
main(title, env, policy, mode=mode, config=config)