Skip to content

Commit e57d5f2

Browse files
guangyeysvekars
andauthored
Update beginner_source/basics/buildmodel_tutorial.py
Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent 617ff9f commit e57d5f2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/basics/buildmodel_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
#############################################
3333
# Get Device for Training
3434
# -----------------------
35-
# We want to be able to train our model on an `accelerator <https://pytorch.org/docs/stable/torch.html#accelerators>`
35+
# We want to be able to train our model on an `accelerator <https://pytorch.org/docs/stable/torch.html#accelerators>`__
3636
# such as CUDA, MPS, MTIA, or XPU. If the current accelerator is available, we will use it. Otherwise, we use the CPU.
3737

3838
device = torch.accelerator.current_accelerator().type if torch.accelerator.is_available() else "cpu"

0 commit comments

Comments
 (0)