-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
27 lines (23 loc) · 645 Bytes
/
main.py
File metadata and controls
27 lines (23 loc) · 645 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
import gym
import gym_sumo
from src.agent.dqn_agent import DQNAgent
from src.agent.dqn_per_agent import DQNPERAgent
def mul_proc_training(process):
agent = DQNPERAgent(process)
#agent = DQNAgent(process)
env = gym.make("sumo-v0", seed="12454"+str(process), render_mode="")
agent.train_RL(env) # for training
import multiprocessing as mp
from multiprocessing import Pool, Process
def main():
ra = [i for i in range(1)]
proces = []
for i in ra:
proc = Process(target=mul_proc_training, args=(str(i+1),))
proces.append(proc)
proc.start()
# complete
for proc in proces:
proc.join()
if __name__ == '__main__':
main()