|
158 | 158 |
|
159 | 159 |
|
160 | 160 | <div class="version">
|
161 |
| - <a href='http://pytorch.org/docs/versions.html'>1.7.0a0+62f4f87 ▼</a> |
| 161 | + <a href='http://pytorch.org/docs/versions.html'>1.7.0a0+dac393f ▼</a> |
162 | 162 | </div>
|
163 | 163 |
|
164 | 164 |
|
@@ -374,7 +374,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
374 | 374 | <span class="k">return</span> <span class="n">generator_context</span>
|
375 | 375 |
|
376 | 376 |
|
377 |
| -<span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 377 | +<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> |
378 | 378 | <span class="sa">r</span><span class="sd">"""Context-manager that disabled gradient calculation.</span>
|
379 | 379 |
|
380 | 380 | <span class="sd"> Disabling gradient calculation is useful for inference, when you are sure</span>
|
@@ -409,10 +409,10 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
409 | 409 | <span class="n">torch</span><span class="o">.</span><span class="n">_C</span><span class="o">.</span><span class="n">set_grad_enabled</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span>
|
410 | 410 |
|
411 | 411 | <span class="k">def</span> <span class="fm">__exit__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">):</span>
|
412 |
| - <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> |
| 412 | + <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> |
413 | 413 |
|
414 | 414 |
|
415 |
| -<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> |
| 415 | +<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> |
416 | 416 | <span class="sa">r</span><span class="sd">"""Context-manager that enables gradient calculation.</span>
|
417 | 417 |
|
418 | 418 | <span class="sd"> Enables gradient calculation, if it has been disabled via :class:`~no_grad`</span>
|
@@ -451,7 +451,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
451 | 451 | <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>
|
452 | 452 |
|
453 | 453 |
|
454 |
| -<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> |
| 454 | +<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> |
455 | 455 | <span class="sa">r</span><span class="sd">"""Context-manager that sets gradient calculation to on or off.</span>
|
456 | 456 |
|
457 | 457 | <span class="sd"> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</span>
|
@@ -493,7 +493,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
493 | 493 | <span class="k">pass</span>
|
494 | 494 |
|
495 | 495 | <span class="k">def</span> <span class="fm">__exit__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">):</span>
|
496 |
| - <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> |
| 496 | + <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> |
497 | 497 | </pre></div>
|
498 | 498 |
|
499 | 499 | </article>
|
|
0 commit comments