Set Training Parameters

Author: Sayed-Hossein-Hosseini

RhythmAttention_Hybrid_CNN_Transformer_Architecture_for_Arrhythmia_Classification.ipynb

@@ -565,6 +565,25 @@
         "    )\n",
         "]"
       ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "## **Set Training Parameters**"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 13,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "# Set Training Parameters\n",
+        "BATCH_SIZE = 64\n",
+        "# Since we use a generator, we define how many steps constitute one epoch\n",
+        "STEPS_PER_EPOCH = len(X_train) // BATCH_SIZE"
+      ]
     }
   ],
   "metadata": {