diff --git a/run.py b/run.py
index 368e10e..9fcda46 100644
--- a/run.py
+++ b/run.py
@@ -5,7 +5,11 @@ torch.set_float32_matmul_precision("medium")
 
 
 def main():
-    LightningCLI(subclass_mode_data=True, subclass_mode_model=True)
+    LightningCLI(
+        subclass_mode_data=True,
+        subclass_mode_model=True,
+        save_config_kwargs={"overwrite": True},
+    )
 
 
 if __name__ == "__main__":
diff --git a/submit.sh b/submit.sh
index 249382f..df46f09 100755
--- a/submit.sh
+++ b/submit.sh
@@ -1,8 +1,5 @@
 #!/bin/bash
-# python run.py fit --config experiments/config_4_layer_8_hidden.yaml
-# python run.py fit --config experiments/config_8_layer_8_hidden.yaml
-python run.py fit --config experiments/config_8_layer_16_hidden_refined.yaml 
-python run.py fit --config experiments/config_16_layer_8_hidden_refined.yaml
-python run.py fit --config experiments/config_16_layer_16_hidden_refined.yaml
-python run.py fit --config experiments/config_8_layer_16_hidden.yaml
-# python run.py fit --config experiments/config_4_layer_16_hidden.yaml
+export CUDA_VISIBLE_DEVICES=1
+python run.py fit --config experiments/5_steps/config_16_layer_16_hidden_refined.yaml
+python run.py fit --config experiments/5_steps/config_32_layer_16_hidden_refined.yaml
+python run.py fit --config experiments/5_steps/config_8_layer_16_hidden_refined.yaml