From f656fa3d2592c92af5f756a96165e7c6c3be3757 Mon Sep 17 00:00:00 2001
From: r-sarma <126173968+r-sarma@users.noreply.github.com>
Date: Fri, 7 Jun 2024 11:36:10 +0200
Subject: [PATCH] Update startscript.sh

---
 use-cases/mnist/tensorflow/startscript.sh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/use-cases/mnist/tensorflow/startscript.sh b/use-cases/mnist/tensorflow/startscript.sh
index 38d2b9b1..bb3122e1 100644
--- a/use-cases/mnist/tensorflow/startscript.sh
+++ b/use-cases/mnist/tensorflow/startscript.sh
@@ -28,6 +28,11 @@ ml Stages/2024 GCC/12.3.0 OpenMPI CUDA/12 MPI-settings/CUDA Python/3.11 HDF5 Pne
 # shellcheck source=/dev/null
 source ~/.bashrc
 
+# Using legacy (2.16) version of Keras
+# Latest version with TF (2.16) installs Keras 3.3
+# which returns an error for multi-node execution
+export TF_USE_LEGACY_KERAS=1
+
 # ON LOGIN NODE download datasets:
 # ../../../.venv-tf/bin/itwinai exec-pipeline --config pipeline.yaml --pipe-key pipeline --steps 0
 source ../../../envAItf_hdfml/bin/activate