diff --git a/HPC_scripts/train_era5.sh b/HPC_scripts/train_era5.sh
index cb53037b01d5471ca7945216692ddb50efc95d4f..2de37e4479b322104ee861d434f4ea65fb4ec4a5 100755
--- a/HPC_scripts/train_era5.sh
+++ b/HPC_scripts/train_era5.sh
@@ -24,5 +24,5 @@ module load cuDNN/7.5.1.10-CUDA-10.1.105
 
 #srun  python scripts/train.py --input_dir data/kth --dataset kth  --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/kth/ours_savp
 
-srun python ../scripts/train_v2.py --input_dir ../data/era5_size_64_64_3_3t_norm --dataset era5  --model savp --model_hparams_dict ../hparams/kth/ours_savp/model_hparams.json --output_dir ../logs/era5_size_64_64_3_3t_norm/ours_savp
+srun python ../scripts/train_v2.py --input_dir ../data/era5_size_64_64_3_3t_norm  --dataset era5  --model savp --model_hparams_dict ../hparams/kth/ours_savp/model_hparams.json --output_dir ../logs/era5_size_64_64_3_3t_norm/ours_savp
 #srun  python scripts/train.py --input_dir data/era5 --dataset era5  --model savp --model_hparams_dict hparams/kth/ours_savp/model_hparams.json --output_dir logs/era5/ours_savp