revert back to .byte() instead of .bool() for PyTorch 1.1

nkolot · Jan 10, 2020 · 21a148d · 21a148d
1 parent b4c70d5
commit 21a148d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train/fits_dict.py b/train/fits_dict.py
@@ -61,7 +61,7 @@ def __setitem__(self, x, val):
 
     def flip_pose(self, pose, is_flipped):
         """flip SMPL pose parameters"""
-        is_flipped = is_flipped.bool()
+        is_flipped = is_flipped.byte()
         pose_f = pose.clone()
         pose_f[is_flipped, :] = pose[is_flipped][:, self.flipped_parts]
         # we also negate the second and the third dimension of the axis-angle representation