forked from LTL2Action/LTL2Action
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexperiments_paper_reach_avoid_policy_gatv2conv_dfa.sh
11 lines (8 loc) · 1.62 KB
/
experiments_paper_reach_avoid_policy_gatv2conv_dfa.sh
1
2
3
4
5
6
7
8
9
10
11
seed=$1
# One-Step Reach-Avoid
python train_agent.py --algo ppo --env Letter-7x7-v3 --log-interval 1 --save-interval 20 --frames 20000000 --discount 0.94 --ltl-sampler CompositionalReachAvoid_1_1_1_5 --epochs 4 --lr 0.0003 --seed ${seed} --dfa --gnn GATv2Conv --pretrained-gnn --pretrained-gnn-sampler CompositionalGeneralDFA --freeze-ltl &> outputs/dfa_train_CompositionalReachAvoid_1_1_1_5_pretrained_GATv2Conv_frozen_seed_${seed}.txt &
python train_agent.py --algo ppo --env Letter-7x7-v3 --log-interval 1 --save-interval 20 --frames 20000000 --discount 0.94 --ltl-sampler CompositionalReachAvoid_1_1_1_5 --epochs 4 --lr 0.0003 --seed ${seed} --dfa --gnn GATv2Conv --pretrained-gnn --pretrained-gnn-sampler CompositionalGeneralDFA &> outputs/dfa_train_CompositionalReachAvoid_1_1_1_5_pretrained_GATv2Conv_seed_${seed}.txt &
# Multi-Step Reach-Avoid
python train_agent.py --algo ppo --env Letter-7x7-v3 --log-interval 1 --save-interval 20 --frames 20000000 --discount 0.94 --ltl-sampler CompositionalReachAvoid_1_5_1_5 --epochs 4 --lr 0.0003 --seed ${seed} --dfa --gnn GATv2Conv --pretrained-gnn --pretrained-gnn-sampler CompositionalGeneralDFA --freeze-ltl &> outputs/dfa_train_CompositionalReachAvoid_1_5_1_5_pretrained_GATv2Conv_frozen_seed_${seed}.txt &
python train_agent.py --algo ppo --env Letter-7x7-v3 --log-interval 1 --save-interval 20 --frames 20000000 --discount 0.94 --ltl-sampler CompositionalReachAvoid_1_5_1_5 --epochs 4 --lr 0.0003 --seed ${seed} --dfa --gnn GATv2Conv --pretrained-gnn --pretrained-gnn-sampler CompositionalGeneralDFA &> outputs/dfa_train_CompositionalReachAvoid_1_5_1_5_pretrained_GATv2Conv_seed_${seed}.txt &
wait