@@ -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 "> "Weight is not supported in sparse mode for now."</ 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 "> 'params'</ 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 "> "Loss:"</ 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 "> "Last Loss:"</ 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 "> "Reject Count:"</ 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 "> "Damping:"</ span > < span  class ="p "> ,</ span >  < span  class ="n "> pg</ span > < span  class ="p "> [</ span > < span  class ="s1 "> 'damping'</ 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 "> <</ 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 "> <</ 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 "> 'params'</ 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