File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change 133
133
# - To mark some parameters in your neural network as **frozen parameters**.
134
134
# - To **speed up computations** when you are only doing forward pass, because computations on tensors that do
135
135
# not track gradients would be more efficient.
136
- # See this `note <https://docs.pytorch.org/docs/stable/notes/autograd.html#locally-disabling-gradient-computation>`_
136
+ # See this `note <https://docs.pytorch.org/docs/stable/notes/autograd.html#locally-disabling-gradient-computation>`_
137
137
# for additional reference.
138
138
139
139
######################################################################
163
163
#
164
164
# To get a sense of what this computational graph looks like we can use the following tools:
165
165
#
166
- # 1. torchviz is a package to visualize computational graphs
167
- # `pytorchviz <https://github.com/szagoruyko/pytorchviz>`_
166
+ #1. torchviz is a package to visualize computational graphs.
167
+ # See the repository here: `https://github.com/szagoruyko/pytorchviz <https://github.com/szagoruyko/pytorchviz>`_
168
+ #
169
+ #2. Setting ``TORCH_LOGS="+autograd"`` enables logging for the backward pass. See details in this
170
+ # discussion: `https://dev-discuss.pytorch.org/t/highlighting-a-few-recent-autograd-features-h2-2023/1787 <https://dev-discuss.pytorch.org/t/highlighting-a-few-recent-autograd-features-h2-2023/1787>`_
168
171
#
169
- # 2. TORCH_LOGS="+autograd" enables logging for the backward pass.
170
- # `<https://dev-discuss.pytorch.org/t/highlighting-a-few-recent-autograd-features-h2-2023/1787>`_
171
172
#
172
173
#
173
174
# .. note::
You can’t perform that action at this time.
0 commit comments