|
158 | 158 |
|
159 | 159 |
|
160 | 160 | <div class="version">
|
161 |
| - <a href='http://pytorch.org/docs/versions.html'>1.7.0a0+83b87de ▼</a> |
| 161 | + <a href='http://pytorch.org/docs/versions.html'>1.7.0a0+0929cd7 ▼</a> |
162 | 162 | </div>
|
163 | 163 |
|
164 | 164 |
|
@@ -376,7 +376,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
376 | 376 | <span class="k">return</span> <span class="n">generator_context</span>
|
377 | 377 |
|
378 | 378 |
|
379 |
| -<div class="viewcode-block" id="no_grad"><a class="viewcode-back" href="../../../autograd.html#torch.autograd.no_grad">[docs]</a><span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 379 | +<div class="viewcode-block" id="no_grad"><a class="viewcode-back" href="../../../generated/torch.no_grad.html#torch.no_grad">[docs]</a><span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
380 | 380 | <span class="sa">r</span><span class="sd">"""Context-manager that disabled gradient calculation.</span>
|
381 | 381 |
|
382 | 382 | <span class="sd"> Disabling gradient calculation is useful for inference, when you are sure</span>
|
@@ -414,7 +414,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
414 | 414 | <span class="n">torch</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span></div>
|
415 | 415 |
|
416 | 416 |
|
417 |
| -<div class="viewcode-block" id="enable_grad"><a class="viewcode-back" href="../../../autograd.html#torch.autograd.enable_grad">[docs]</a><span class="k">class</span> <span class="nc">enable_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 417 | +<div class="viewcode-block" id="enable_grad"><a class="viewcode-back" href="../../../generated/torch.enable_grad.html#torch.enable_grad">[docs]</a><span class="k">class</span> <span class="nc">enable_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
418 | 418 | <span class="sa">r</span><span class="sd">"""Context-manager that enables gradient calculation.</span>
|
419 | 419 |
|
420 | 420 | <span class="sd"> Enables gradient calculation, if it has been disabled via :class:`~no_grad`</span>
|
@@ -453,7 +453,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
453 | 453 | <span class="n">torch</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span></div>
|
454 | 454 |
|
455 | 455 |
|
456 |
| -<div class="viewcode-block" id="set_grad_enabled"><a class="viewcode-back" href="../../../autograd.html#torch.autograd.set_grad_enabled">[docs]</a><span class="k">class</span> <span class="nc">set_grad_enabled</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
| 456 | +<div class="viewcode-block" id="set_grad_enabled"><a class="viewcode-back" href="../../../generated/torch.set_grad_enabled.html#torch.set_grad_enabled">[docs]</a><span class="k">class</span> <span class="nc">set_grad_enabled</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
457 | 457 | <span class="sa">r</span><span class="sd">"""Context-manager that sets gradient calculation to on or off.</span>
|
458 | 458 |
|
459 | 459 | <span class="sd"> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</span>
|
|
0 commit comments