-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathpaper_experiments.ps1
70 lines (51 loc) · 3.15 KB
/
paper_experiments.ps1
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
$RUNS = 5
# This script runs 5 * 10 = 50 experiments so it will run for a REALLY LONG time.
# It is meant to be used to replicate all the experiments in the paper.
for($i = 1; $i -le $RUNS; $i++){
# Sokoban Experiments
$GAME = "sokoban"
Write-Output "${i}/${RUNS}: $GAME (Base Experiment) ..."
$PATH = ".\experiments\$GAME\GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\base.yml -ovr save_path=$PATH
./analyze.ps1 $GAME $PATH
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Push Signature) ..."
$PATH = ".\experiments\$GAME\DIVPUSHSIG_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\div-pushsig.yml -ovr save_path=$PATH
./analyze.ps1 $GAME $PATH
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple) ..."
$PATH = ".\experiments\$GAME\DIV_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\div-tuple.yml -ovr save_path=$PATH dataset_config.property_reward:=None dataset_config.data_augmentation:=False
./analyze.ps1 $GAME $PATH
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Property Reward) ..."
$PATH = ".\experiments\$GAME\DIV_PR_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\div-tuple.yml -ovr save_path=$PATH dataset_config.data_augmentation:=False
./analyze.ps1 $GAME $PATH
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Data Augmentation) ..."
$PATH = ".\experiments\$GAME\DIV_AUG_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\div-tuple.yml -ovr save_path=$PATH dataset_config.property_reward:=None
./analyze.ps1 $GAME $PATH
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Property Reward + Data Augmentation) ..."
$PATH = ".\experiments\$GAME\DIV_PR_AUG_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\div-tuple.yml -ovr save_path=$PATH
./analyze.ps1 $GAME $PATH
# Zelda Experiments
$GAME = "zelda"
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Property Reward) ..."
$PATH = ".\experiments\$GAME\DIV_PR_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\config.yml -ovr save_path=$PATH dataset_config.data_augmentation:=False
./analyze.ps1 $GAME $PATH -TAILOREDGMM
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Property Reward + Data Augmentation) ..."
$PATH = ".\experiments\$GAME\DIV_PR_AUG_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\config.yml -ovr save_path=$PATH
./analyze.ps1 $GAME $PATH -TAILOREDGMM
# Danger Dave Experiments
$GAME = "dave"
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple) ..."
$PATH = ".\experiments\$GAME\DIV_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\config.yml -ovr save_path=$PATH dataset_config.data_augmentation:=False
./analyze.ps1 $GAME $PATH -TAILOREDGMM
Write-Output "${i}/${RUNS}: $GAME (Diversity Sampling: Tuple + Data Augmentation) ..."
$PATH = ".\experiments\$GAME\DIV_AUG_GFLOW\${i}"
python .\cli.py train -cfg .\configs\$GAME\gflownet\config.yml -ovr save_path=$PATH
./analyze.ps1 $GAME $PATH -TAILOREDGMM
}