Final Experiment Configs

Author: david-baek

notebooks/plot_runs.ipynb

@@ -3,8 +3,8 @@
 #SBATCH --gres=gpu:a100:1
 #SBATCH -n 16
 
-for ARG in $(python -c "import numpy as np; print(' '.join(map(str, np.linspace(49, 100, 20, dtype=int))))"); do
+for ARG in $(python -c "import numpy as np; print(' '.join(map(str, np.linspace(51, 100, 20, dtype=int))))"); do
     echo "Running with seed $ARG:"
-    python ../src/unit_exp.py --data_id circle --model_id H_transformer --seed $ARG
+    python ../src/unit_exp.py --data_id circle --model_id H_MLP --seed $ARG
     echo
 done

scripts/u_circle.sh

@@ -1,10 +1,10 @@
 #!/bin/bash
 #SBATCH -t 16:00:00
-#SBATCH --gres=gpu:1
+#SBATCH --gres=gpu:a100:1
 #SBATCH -n 16
 
-python ../src/unit_exp.py --data_id family_tree --model_id standard_transformer
-python ../src/unit_exp.py --data_id family_tree --model_id H_transformer
-python ../src/unit_exp.py --data_id family_tree --model_id standard_MLP
-python ../src/unit_exp.py --data_id family_tree --model_id H_MLP
-
+for ARG in $(python -c "import numpy as np; print(' '.join(map(str, np.linspace(51, 100, 20, dtype=int))))"); do
+    echo "Running with seed $ARG:"
+    python ../src/unit_exp.py --data_id family_tree --model_id H_MLP --seed $ARG
+    echo
+done

scripts/u_family.sh

@@ -19,7 +19,7 @@
 model_id_choices = ["H_MLP", "standard_MLP", "H_transformer", "standard_transformer"]
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Experiment')
-    parser.add_argument('--seed', type=int, default=49, help='random seed')
+    parser.add_argument('--seed', type=int, default=66, help='random seed')
     parser.add_argument('--data_id', type=str, required=True, choices=data_id_choices, help='Data ID')
     parser.add_argument('--model_id', type=str, required=True, choices=model_id_choices, help='Model ID')