Skip to content

Commit 46908e8

Browse files
committed
Add docs from pypose/pypose
1 parent 50b44b7 commit 46908e8

File tree

1 file changed

+1
-1
lines changed
  • docs/preview/pr/378/_modules/pypose/optim/optimizer

1 file changed

+1
-1
lines changed

docs/preview/pr/378/_modules/pypose/optim/optimizer/index.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -853,6 +853,7 @@ <h1>Source code for pypose.optim.optimizer</h1><div class="highlight"><pre>
853853

854854
<span class="k">def</span><span class="w"> </span><span class="nf">_step_sparse</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="nb">input</span><span class="p">,</span> <span class="n">target</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">weight</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
855855
<span class="c1"># Note: sparse mode currently assumes a single residual and does not support weights.</span>
856+
<span class="k">assert</span> <span class="n">weight</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">,</span> <span class="s2">&quot;Weight is not supported in sparse mode for now.&quot;</span>
856857
<span class="k">for</span> <span class="n">pg</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">param_groups</span><span class="p">:</span>
857858
<span class="c1"># weight is not used in sparse mode for now.</span>
858859
<span class="c1"># weight = self.weight if weight is None else weight</span>
@@ -887,7 +888,6 @@ <h1>Source code for pypose.optim.optimizer</h1><div class="highlight"><pre>
887888
<span class="k">break</span>
888889
<span class="bp">self</span><span class="o">.</span><span class="n">update_parameter</span><span class="p">(</span><span class="n">pg</span><span class="p">[</span><span class="s1">&#39;params&#39;</span><span class="p">],</span> <span class="n">D</span><span class="p">)</span>
889890
<span class="bp">self</span><span class="o">.</span><span class="n">loss</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">model</span><span class="o">.</span><span class="n">loss</span><span class="p">(</span><span class="nb">input</span><span class="p">,</span> <span class="n">target</span><span class="p">)</span>
890-
<span class="nb">print</span><span class="p">(</span><span class="s2">&quot;Loss:&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">loss</span><span class="p">,</span> <span class="s2">&quot;Last Loss:&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">last</span><span class="p">,</span> <span class="s2">&quot;Reject Count:&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">reject_count</span><span class="p">,</span> <span class="s2">&quot;Damping:&quot;</span><span class="p">,</span> <span class="n">pg</span><span class="p">[</span><span class="s1">&#39;damping&#39;</span><span class="p">])</span>
891891
<span class="bp">self</span><span class="o">.</span><span class="n">strategy</span><span class="o">.</span><span class="n">update</span><span class="p">(</span><span class="n">pg</span><span class="p">,</span> <span class="n">last</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">last</span><span class="p">,</span> <span class="n">loss</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">loss</span><span class="p">,</span> <span class="n">J</span><span class="o">=</span><span class="n">J</span><span class="p">,</span> <span class="n">D</span><span class="o">=</span><span class="n">D</span><span class="p">,</span> <span class="n">R</span><span class="o">=</span><span class="n">R</span><span class="o">.</span><span class="n">view</span><span class="p">(</span><span class="o">-</span><span class="mi">1</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span>
892892
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">last</span> <span class="o">&lt;</span> <span class="bp">self</span><span class="o">.</span><span class="n">loss</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">reject_count</span> <span class="o">&lt;</span> <span class="bp">self</span><span class="o">.</span><span class="n">reject</span><span class="p">:</span> <span class="c1"># reject step</span>
893893
<span class="bp">self</span><span class="o">.</span><span class="n">update_parameter</span><span class="p">(</span><span class="n">params</span><span class="o">=</span><span class="n">pg</span><span class="p">[</span><span class="s1">&#39;params&#39;</span><span class="p">],</span> <span class="n">step</span><span class="o">=-</span><span class="n">D</span><span class="p">)</span>

0 commit comments

Comments
 (0)