Change the max_steps from 5 to 1

Nixtla · Mar 1, 2024 · 6064449 · 6064449
1 parent 8d6f423
commit 6064449
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nbs/core.ipynb b/nbs/core.ipynb
@@ -2499,7 +2499,7 @@
     "    params = {\n",
     "        \"h\": 12, \n",
     "        \"input_size\": 24, \n",
-    "        \"max_steps\": 5, \n",
+    "        \"max_steps\": 1, \n",
     "        \"optimizer\": torch.optim.Adadelta, \n",
     "        \"optimizer_kwargs\": {\"lr\": 0.8, \"rho\": 0.45}\n",
     "    }\n",