vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: remove orientation optimizer from pose

Browse Source
This commit is contained in:
Wlad 2021-02-21 23:58:02 +01:00
parent b18e54c397
commit 37aa619337
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train_pose.py
View File

@ -85,7 +85,8 @@ def train_pose(
pose_layer = BodyPose(model, dtype=dtype, device=device,
useBodyMeanAngles=False).to(device=device, dtype=dtype)
parameters = [model.global_orient, pose_layer.body_pose]
# [model.global_orient, pose_layer.body_pose]
parameters = [pose_layer.body_pose] # list(model.parameters())
# setup all loss layers
for l in extra_loss_layers: