huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

adding final commands for tuning, training final models, testing them

Browse Source
This commit is contained in:
Arec Jamgochian
2022-03-03 12:12:12 -08:00
parent f5f1c24f45
commit fc04f8e9ee
2 changed files with 39 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
train_models.sh Executable file
View File

@@ -0,0 +1,28 @@
# Hyperparameter tuning runs (will run search defined in files)
# Can add --epochs kwarg to change number of training epochs
# shail-experiment.py will generate the hail config aswell
# Experiment A
python bc-experiment.py --train A
python gail-experiment.py --train A
python shail-experiment.py --train A
# Experiment B
python bc-experiment.py --train B
python gail-experiment.py --train B
python shail-experiment.py --train B
# Retrain best models with 5 seeds
# Can add --test_seeds and --test_cpus kwargs to change number of seeds, allocate more memory
# Experiment A
python bc-experiment.py --test best_configs/bc_expA.json
python gail-experiment.py --test best_configs/gail_expA.json
python shail-experiment.py --train best_configs/hail_expA.json
python shail-experiment.py --train best_configs/shail_expA.json
# Experiment B
python bc-experiment.py --test best_configs/bc_expB.json
python gail-experiment.py --test best_configs/gail_expB.json
python shail-experiment.py --train best_configs/hail_expB.json
python shail-experiment.py --train best_configs/shail_expB.json