forked from IntelLabs/networkgym
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtrain_online_parallel.sh
41 lines (36 loc) · 1.66 KB
/
train_online_parallel.sh
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
30
31
32
33
34
35
36
37
38
39
40
41
#!/bin/bash
# NOTE: trains PPO and SAC on parallel envs
# echo "Cleaning up broken environments"
# python -u NetworkAgent/config_changer.py --test --agent PPO --seed 0 --steps 2
# cd NetworkAgent/stable-baselines3
# python -u main_rl.py --env nqos_split --client_id 0
# python -u main_rl.py --env nqos_split --client_id 1
# python -u main_rl.py --env nqos_split --client_id 2
# python -u main_rl.py --env nqos_split --client_id 3
# python -u main_rl.py --env nqos_split --client_id 4
# python -u main_rl.py --env nqos_split --client_id 5
# python -u main_rl.py --env nqos_split --client_id 6
# python -u main_rl.py --env nqos_split --client_id 7
# cd ../..
# echo "Training PPO"
# python -u NetworkAgent/config_changer.py --train --parallel_env --agent PPO --seed 16 --steps 80000
# cd NetworkAgent/stable-baselines3
# python -u main_rl.py --env nqos_split --client_id 0
# cd ../..
# echo "Cleaning up broken environments"
# python -u NetworkAgent/config_changer.py --test --agent PPO --seed 0 --steps 2
# cd NetworkAgent/stable-baselines3
# python -u main_rl.py --env nqos_split --client_id 0
# python -u main_rl.py --env nqos_split --client_id 1
# python -u main_rl.py --env nqos_split --client_id 2
# python -u main_rl.py --env nqos_split --client_id 3
# python -u main_rl.py --env nqos_split --client_id 4
# python -u main_rl.py --env nqos_split --client_id 5
# python -u main_rl.py --env nqos_split --client_id 6
# python -u main_rl.py --env nqos_split --client_id 7
# cd ../..
echo "Training SAC"
python -u NetworkAgent/config_changer.py --train --parallel_env --agent SAC --seed 16 --steps 80000
cd NetworkAgent/stable-baselines3
python -u main_rl.py --env nqos_split --client_id 0
cd ../..