diff --git a/HW02/HW02.ipynb b/HW02/HW02.ipynb
index 1a1d4dab..ae888b7d 100644
--- a/HW02/HW02.ipynb
+++ b/HW02/HW02.ipynb
@@ -371,20 +371,21 @@
       "cell_type": "code",
       "source": [
         "# data prarameters\n",
-        "concat_nframes = 1              # the number of frames to concat with, n must be odd (total 2k+1 = n frames)\n",
+        "concat_nframes = 19              # the number of frames to concat with, n must be odd (total 2k+1 = n frames)\n",
         "train_ratio = 0.8               # the ratio of data used for training, the rest will be used for validation\n",
         "\n",
         "# training parameters\n",
         "seed = 0                        # random seed\n",
-        "batch_size = 512                # batch size\n",
+        "batch_size = 2048               # batch size\n",
         "num_epoch = 5                   # the number of training epoch\n",
+        "early_stopping=8
         "learning_rate = 0.0001          # learning rate\n",
         "model_path = './model.ckpt'     # the path where the checkpoint will be saved\n",
         "\n",
         "# model parameters\n",
         "input_dim = 39 * concat_nframes # the input dim of the model, you should not change the value\n",
-        "hidden_layers = 1               # the number of hidden layers\n",
-        "hidden_dim = 256                # the hidden dim"
+        "hidden_layers = 3               # the number of hidden layers\n",
+        "hidden_dim = 1024                # the hidden dim"
       ],
       "metadata": {
         "id": "iIHn79Iav1ri"
@@ -958,4 +959,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}