@@ -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