Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
0d22412
1
Parent(s):
bb70c42
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 |
{
|