diff --git a/.gitignore b/.gitignore
index f63db40b29ed4c0a3aefc3e66fb9048a80859ce9..fc1b5337b8aa879471c7f0e9e12fe880f6c5135c 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,6 @@ __pycache__/*
 *.subset
 *.pyc
 wandb/*
+job.sbatch
+logslurms/*
+
diff --git a/job.py b/job.py
index 6c95bd20613598988b226549d4f809f388893690..757d00d67386634c613b628ae68bc7dcd01d42c4 100644
--- a/job.py
+++ b/job.py
@@ -37,7 +37,7 @@ source venv/bin/activate
 python -m pip install -r requirements.txt
 
 echo "Running main.py"
-python3 main.py --logDir /usr/users/sdi1/sdi1_3/Projet_DL/Kaggle_Phytoplankton/logs/ --no_wandb
+python3 main.py --rootDir /usr/users/sdi1/sdi1_3/Projet_DL/Kaggle_Phytoplankton/logs/ --no_wandb
 
 if [[ $? != 0 ]]; then
     exit -1