Add th_comps to buffers

hassony2 · Sep 10, 2020 · 4f1dcad · 4f1dcad
1 parent c187374
commit 4f1dcad
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/manopth/manolayer.py b/manopth/manolayer.py
@@ -94,6 +94,7 @@ def __init__(self,
             # Save as axis-angle
             self.register_buffer('th_hands_mean', th_hands_mean)
             selected_components = hands_components[:ncomps]
+            self.register_buffer('th_comps', torch.Tensor(hands_components))
             self.register_buffer('th_selected_comps',
                                  torch.Tensor(selected_components))
         else: