removed temp params

hkproj · May 22, 2023 · f1bba48 · f1bba48
1 parent e323da5
commit f1bba48
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Local_Train.ipynb b/Local_Train.ipynb
@@ -72,9 +72,9 @@
       "source": [
         "from config import get_config\n",
         "cfg = get_config()\n",
-        "cfg['batch_size'] = 8\n",
-        "cfg['preload'] = '29'\n",
-        "cfg['num_epochs'] = 40\n",
+        "cfg['batch_size'] = 6\n",
+        "cfg['preload'] = None\n",
+        "cfg['num_epochs'] = 30\n",
         "\n",
         "from train import train_model\n",
         "\n",