Reduce length of DL part of tutorial

examples/pysr_demo.ipynb

@@ -1064,7 +1064,7 @@
       "outputs": [],
       "source": [
         "hidden = 128\n",
-        "total_steps = 50000\n",
+        "total_steps = 10_000\n",
         "\n",
         "\n",
         "def mlp(size_in, size_out, act=nn.ReLU):\n",
@@ -1284,7 +1284,7 @@
         "np.random.seed(1)\n",
         "tmpX = X_for_pysr.detach().numpy().reshape(-1, 5)\n",
         "tmpy = y_i_for_pysr.detach().numpy().reshape(-1)\n",
-        "idx2 = np.random.randint(0, tmpy.shape[0], size=3000)\n",
+        "idx2 = np.random.randint(0, tmpy.shape[0], size=500)\n",
         "\n",
         "model = PySRRegressor(\n",
         "    niterations=20,\n",