MilesCranmer commited on
Commit
0d22412
1 Parent(s): bb70c42

Update deprecated lightning call

Browse files
Files changed (1) hide show
  1. examples/pysr_demo.ipynb +3 -1
examples/pysr_demo.ipynb CHANGED
@@ -1206,7 +1206,9 @@
1206
  },
1207
  "outputs": [],
1208
  "source": [
1209
- "trainer = pl.Trainer(max_steps=total_steps, gpus=1, benchmark=True)"
 
 
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
  {