Spaces:
Running
Running
Update deprecated lightning call
Browse files- examples/pysr_demo.ipynb +3 -1
examples/pysr_demo.ipynb
CHANGED
|
@@ -1206,7 +1206,9 @@
|
|
| 1206 |
},
|
| 1207 |
"outputs": [],
|
| 1208 |
"source": [
|
| 1209 |
-
"trainer = pl.Trainer(
|
|
|
|
|
|
|
| 1210 |
]
|
| 1211 |
},
|
| 1212 |
{
|
|
|
|
| 1206 |
},
|
| 1207 |
"outputs": [],
|
| 1208 |
"source": [
|
| 1209 |
+
"trainer = pl.Trainer(\n",
|
| 1210 |
+
" max_steps=total_steps, accelerator=\"gpu\", devices=1, benchmark=True\n",
|
| 1211 |
+
")\n"
|
| 1212 |
]
|
| 1213 |
},
|
| 1214 |
{
|