From e220cbe53b6bb90818a88e8c07c922e9e4a0a4d1 Mon Sep 17 00:00:00 2001 From: Fahad Khalid <f.khalid@fz-juelich.de> Date: Tue, 18 May 2021 18:08:03 +0200 Subject: [PATCH] Job scripts for Tensorflow now use the keras_mnist example by default. --- tensorflow/jureca_job.sh | 2 +- tensorflow/jusuf_job.sh | 2 +- tensorflow/juwels_booster_job.sh | 2 +- tensorflow/juwels_job.sh | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tensorflow/jureca_job.sh b/tensorflow/jureca_job.sh index e818bc0..864c5ec 100755 --- a/tensorflow/jureca_job.sh +++ b/tensorflow/jureca_job.sh @@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5 export CUDA_VISIBLE_DEVICES=0,1,2,3 # Run the program -srun python -u mnist.py +srun python -u keras_mnist.py diff --git a/tensorflow/jusuf_job.sh b/tensorflow/jusuf_job.sh index 24f3c83..1c957b6 100755 --- a/tensorflow/jusuf_job.sh +++ b/tensorflow/jusuf_job.sh @@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5 export CUDA_VISIBLE_DEVICES=0 # Run the program -srun python -u mnist.py +srun python -u keras_mnist.py diff --git a/tensorflow/juwels_booster_job.sh b/tensorflow/juwels_booster_job.sh index df9cdef..f8b2b22 100755 --- a/tensorflow/juwels_booster_job.sh +++ b/tensorflow/juwels_booster_job.sh @@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5 export CUDA_VISIBLE_DEVICES=0,1,2,3 # Run the program -srun python -u mnist.py +srun python -u keras_mnist.py diff --git a/tensorflow/juwels_job.sh b/tensorflow/juwels_job.sh index 55831d0..7cae2f5 100755 --- a/tensorflow/juwels_job.sh +++ b/tensorflow/juwels_job.sh @@ -21,4 +21,4 @@ module load Horovod/0.20.3-Python-3.8.5 export CUDA_VISIBLE_DEVICES=0,1,2,3 # Run the program -srun python -u mnist.py +srun python -u keras_mnist.py -- GitLab