diff --git a/keras/4.4-overfitting-and-underfitting.ipynb b/keras/4.4-overfitting-and-underfitting.ipynb
index 06bf4be..c497fa6 100644
--- a/keras/4.4-overfitting-and-underfitting.ipynb
+++ b/keras/4.4-overfitting-and-underfitting.ipynb
@@ -23,7 +23,7 @@
     }
    ],
    "source": [
-    "%env SPARK_DRIVER_MEMORY=14g\n",
+    "%env SPARK_DRIVER_MEMORY=32g\n",
     "%env PYSPARK_PYTHON=/usr/bin/python3.5\n",
     "%env PYSPARK_DRIVER_PYTHON=/usr/bin/python3.5\n",
     "\n",