Suggestion on page 444 #125
baifanhorst
started this conversation in
General
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
At the bottom of page 444, a pytorch lightning trainer class instance is created as:
if torch.cuda.is_available():
trainer = pl.Trainer(max_epochs=10, gpus=1)
else:
trainer = pl.Trainer(max_epochs=10)
However, in the latest version, it should be changed to
if torch.cuda.is_available():
trainer = pl.Trainer(max_epochs=10, accelerator="gpu", devices=1)
else:
trainer = pl.Trainer(max_epochs=10)
Beta Was this translation helpful? Give feedback.
All reactions