-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.sh
10 lines (10 loc) · 1.14 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
#!/bin/bash
# Smoke Plume
python3 run_model.py --dataset=PhiFlow --relaxed_symmetry=Translation --hidden_dim=128 --num_layers=5 --out_length=6 --num_banks=2 --alpha=0 --batch_size=8 --learning_rate=0.001 --decay_rate=0.95 &
python3 run_model.py --dataset=PhiFlow --relaxed_symmetry=Scale --hidden_dim=64 --num_layers=5 --out_length=6 --alpha=1e-6 --batch_size=8 --learning_rate=0.0001 --decay_rate=0.95 &
python3 run_model.py --dataset=PhiFlow --relaxed_symmetry=Rotation --hidden_dim=92 --num_layers=5 --out_length=6 --alpha=1e-5 --batch_size=16 --learning_rate=0.001 --decay_rate=0.95 &
#JetFlow
python3 run_model.py --dataset=JetFlow --relaxed_symmetry=Translation --hidden_dim=128 --num_layers=5 --out_length=6 --num_banks=3 --alpha=1e-5 --batch_size=8 --learning_rate=0.001 --decay_rate=0.9 &
python3 run_model.py --dataset=JetFlow --relaxed_symmetry=Rotation --hidden_dim=16 --num_layers=5 --out_length=6 --alpha=1e-4 --batch_size=16 --learning_rate=0.001 --decay_rate=0.95 &
python3 run_model.py --dataset=JetFlow --relaxed_symmetry=Scale --hidden_dim=32 --num_layers=5 --out_length=7 --alpha=1e-6 --batch_size=8 --learning_rate=0.0001 --decay_rate=0.9 &
wait