Spaces:
Running
Running
Commit
•
7de1c20
1
Parent(s):
7055fde
Reduce NN training steps in example
Browse files- examples/pysr_demo.ipynb +1 -1
examples/pysr_demo.ipynb
CHANGED
|
@@ -1173,7 +1173,7 @@
|
|
| 1173 |
"import pytorch_lightning as pl\n",
|
| 1174 |
"\n",
|
| 1175 |
"hidden = 128\n",
|
| 1176 |
-
"total_steps =
|
| 1177 |
"\n",
|
| 1178 |
"\n",
|
| 1179 |
"def mlp(size_in, size_out, act=nn.ReLU):\n",
|
|
|
|
| 1173 |
"import pytorch_lightning as pl\n",
|
| 1174 |
"\n",
|
| 1175 |
"hidden = 128\n",
|
| 1176 |
+
"total_steps = 50_000\n",
|
| 1177 |
"\n",
|
| 1178 |
"\n",
|
| 1179 |
"def mlp(size_in, size_out, act=nn.ReLU):\n",
|