diff --git a/src/training/sweep_data_augmentation_2.sh b/src/training/sweep_data_augmentation_2.sh
index 6f4a5f4..2bf9989 100755
--- a/src/training/sweep_data_augmentation_2.sh
+++ b/src/training/sweep_data_augmentation_2.sh
@@ -1,3 +1,3 @@
-python -m salience_analysis.few_shot_training --seed=46 --data_augmentation="true"
-python -m salience_analysis.few_shot_training --seed=47 --data_augmentation="true"
-python -m salience_analysis.few_shot_training --seed=48 --data_augmentation="true"
\ No newline at end of file
+python few_shot_training.py  --seed=46 --data_augmentation="true"
+python few_shot_training.py  --seed=47 --data_augmentation="true"
+python few_shot_training.py  --seed=48 --data_augmentation="true"
\ No newline at end of file
diff --git a/src/training/sweep_data_augmentation_3.sh b/src/training/sweep_data_augmentation_3.sh
index c8b7095..7f51299 100755
--- a/src/training/sweep_data_augmentation_3.sh
+++ b/src/training/sweep_data_augmentation_3.sh
@@ -1,3 +1,3 @@
-python -m salience_analysis.few_shot_training --seed=49 --data_augmentation="true"
-python -m salience_analysis.few_shot_training --seed=50 --data_augmentation="true"
-python -m salience_analysis.few_shot_training --seed=51 --data_augmentation="true"
\ No newline at end of file
+python few_shot_training.py  --seed=49 --data_augmentation="true"
+python few_shot_training.py  --seed=50 --data_augmentation="true"
+python few_shot_training.py  --seed=51 --data_augmentation="true"
\ No newline at end of file
diff --git a/src/training/sweep_no_data_augmentation.sh b/src/training/sweep_no_data_augmentation.sh
index 41adfaf..476e4a6 100755
--- a/src/training/sweep_no_data_augmentation.sh
+++ b/src/training/sweep_no_data_augmentation.sh
@@ -1,10 +1,10 @@
-python -m salience_analysis.few_shot_training --seed=42
-python -m salience_analysis.few_shot_training --seed=43
-python -m salience_analysis.few_shot_training --seed=44
-python -m salience_analysis.few_shot_training --seed=45
-python -m salience_analysis.few_shot_training --seed=46
-python -m salience_analysis.few_shot_training --seed=47
-python -m salience_analysis.few_shot_training --seed=48
-python -m salience_analysis.few_shot_training --seed=49
-python -m salience_analysis.few_shot_training --seed=50
-python -m salience_analysis.few_shot_training --seed=51
+python few_shot_training.py  --seed=42
+python few_shot_training.py  --seed=43
+python few_shot_training.py  --seed=44
+python few_shot_training.py  --seed=45
+python few_shot_training.py  --seed=46
+python few_shot_training.py  --seed=47
+python few_shot_training.py  --seed=48
+python few_shot_training.py  --seed=49
+python few_shot_training.py  --seed=50
+python few_shot_training.py  --seed=51