diff --git a/cifar_100_EfficientNet.ipynb b/cifar_100_EfficientNet.ipynb
index 6120d08..3a16326 100644
--- a/cifar_100_EfficientNet.ipynb
+++ b/cifar_100_EfficientNet.ipynb
@@ -448,7 +448,7 @@
     }
    },
    "source": [
-    "We take only 50 samples in each category"
+    "We take only 20 samples in each category"
    ]
   },
   {
@@ -465,7 +465,7 @@
    },
    "outputs": [],
    "source": [
-    "num_samples = 50\n",
+    "num_samples  = 20\n",
     "X_sub, y_sub = subsample_train(X_train, y_train, num_samples)"
    ]
   },