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 dd47349 commit 1551bc3Copy full SHA for 1551bc3
intermediate_source/torchrec_interactive_tutorial.py
@@ -837,7 +837,7 @@ def _wait_impl(self) -> torch.Tensor:
837
# depending on your preferences:
838
# 1. Passing optimizer kwargs through ``fused_params`` in sharder
839
# 2. Through ``apply_optimizer_in_backward``, which converts the optimizer
840
-# parameters to ``fused_params`` to pass to the `TBE`` in the ``EmbeddingBagCollection`` or ``EmbeddingCollection``.
+# parameters to ``fused_params`` to pass to the ``TBE`` in the ``EmbeddingBagCollection`` or ``EmbeddingCollection``.
841
#
842
843
# Approach 1: passing optimizer kwargs through fused parameters
0 commit comments