|
187 | 187 |
|
188 | 188 |
|
189 | 189 | <div class="version">
|
190 |
| - <a href='http://pytorch.org/docs/versions.html'>1.9.0a0+6c8fea9 ▼</a> |
| 190 | + <a href='http://pytorch.org/docs/versions.html'>1.9.0a0+486e771 ▼</a> |
191 | 191 | </div>
|
192 | 192 |
|
193 | 193 |
|
@@ -464,7 +464,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
464 | 464 | <span class="k">raise</span> <span class="ne">NotImplementedError</span>
|
465 | 465 |
|
466 | 466 |
|
467 |
| -<span class="k">class</span> <span class="nc">no_grad</span><span class="p">(</span><span class="n">_DecoratorContextManager</span><span class="p">):</span> |
| 467 | +<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> |
468 | 468 | <span class="sa">r</span><span class="sd">"""Context-manager that disabled gradient calculation.</span>
|
469 | 469 |
|
470 | 470 | <span class="sd"> Disabling gradient calculation is useful for inference, when you are sure</span>
|
@@ -504,7 +504,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
504 | 504 | <span class="n">torch</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>
|
505 | 505 |
|
506 | 506 | <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="n">exc_type</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">exc_value</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">traceback</span><span class="p">:</span> <span class="n">Any</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
507 |
| - <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> |
| 507 | + <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> |
508 | 508 |
|
509 | 509 |
|
510 | 510 | <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>
|
@@ -546,7 +546,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
546 | 546 | <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="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span></div>
|
547 | 547 |
|
548 | 548 |
|
549 |
| -<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> |
| 549 | +<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> |
550 | 550 | <span class="sa">r</span><span class="sd">"""Context-manager that sets gradient calculation to on or off.</span>
|
551 | 551 |
|
552 | 552 | <span class="sd"> ``set_grad_enabled`` will enable or disable grads based on its argument :attr:`mode`.</span>
|
@@ -588,7 +588,7 @@ <h1>Source code for torch.autograd.grad_mode</h1><div class="highlight"><pre>
|
588 | 588 | <span class="k">pass</span>
|
589 | 589 |
|
590 | 590 | <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="n">exc_type</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">exc_value</span><span class="p">:</span> <span class="n">Any</span><span class="p">,</span> <span class="n">traceback</span><span class="p">:</span> <span class="n">Any</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span>
|
591 |
| - <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="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span></div> |
| 591 | + <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="bp">self</span><span class="o">.</span><span class="n">prev</span><span class="p">)</span> |
592 | 592 | </pre></div>
|
593 | 593 |
|
594 | 594 | </article>
|
|
0 commit comments