Update run-keras-notebook.sh

Le-Zheng · May 29, 2019 · 7dc5eb9 · 7dc5eb9
1 parent 33bd036
commit 7dc5eb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras/run-keras-notebook.sh b/keras/run-keras-notebook.sh
@@ -49,10 +49,10 @@ sed -i '/^matplotlib.use('\''Agg'\'')$/a\from matplotlib import pyplot as plt' $
 ${SPARK_HOME}/bin/spark-submit \
         --master ${MASTER} \
         --driver-cores 2  \
-        --driver-memory 16g  \
+        --driver-memory 17g  \
         --total-executor-cores 2  \
         --executor-cores 2  \
-        --executor-memory 16g \
+        --executor-memory 17g \
         --conf spark.akka.frameSize=64 \
         --py-files ${ANALYTICS_ZOO_PYZIP},${ZOO_TUTORIALS}//keras/tmp_test.py  \
         --properties-file ${ANALYTICS_ZOO_CONF} \