From 352ae64865dd0646fff219b9484accf7126b901a Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Thu, 3 Oct 2024 15:04:50 -0700 Subject: [PATCH] Fix the underline that is too short in torchrec_intro_tutorial.py (#3074) * Update torchrec_intro_tutorial.py * Update torchrec_intro_tutorial.py --- intermediate_source/torchrec_intro_tutorial.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/intermediate_source/torchrec_intro_tutorial.py b/intermediate_source/torchrec_intro_tutorial.py index 75d2532670..c4fdfc1833 100644 --- a/intermediate_source/torchrec_intro_tutorial.py +++ b/intermediate_source/torchrec_intro_tutorial.py @@ -39,6 +39,12 @@ # !pip3 install fbgemm_gpu --index-url https://download.pytorch.org/whl/cu121 # !pip3 install torchmetrics==1.0.3 # !pip3 install torchrec --index-url https://download.pytorch.org/whl/cu121 +# +# .. note:: +# If you are running this in Google Colab, make sure to switch to a GPU runtime type. +# For more information, +# see `Enabling CUDA `__ +# @@ -217,7 +223,7 @@ ###################################################################### # TorchRec Modules and Data Types -# ------------------------------ +# ---------------------------------- # # This section goes over TorchRec Modules and data types including such # entities as ``EmbeddingCollection`` and ``EmbeddingBagCollection``,