We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 617ff9f commit e57d5f2Copy full SHA for e57d5f2
beginner_source/basics/buildmodel_tutorial.py
@@ -32,7 +32,7 @@
32
#############################################
33
# Get Device for Training
34
# -----------------------
35
-# We want to be able to train our model on an `accelerator <https://pytorch.org/docs/stable/torch.html#accelerators>`
+# We want to be able to train our model on an `accelerator <https://pytorch.org/docs/stable/torch.html#accelerators>`__
36
# such as CUDA, MPS, MTIA, or XPU. If the current accelerator is available, we will use it. Otherwise, we use the CPU.
37
38
device = torch.accelerator.current_accelerator().type if torch.accelerator.is_available() else "cpu"
0 commit comments