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 minor runtime error

Browse Source
This commit is contained in:
Wlad 2021-02-17 20:53:39 +01:00
parent 6da9a47495
commit 3e40ff67ad
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.yaml
View File

@ -52,7 +52,7 @@ pose:
vposerPath: "./vposer_v1_0" vposerPath: "./vposer_v1_0"
temporal: temporal:
enabled: true enabled: true
iterations: 75 iterations: 50
lr: 0.02 lr: 0.02
preview: preview:
enable: true, enable: true,

2
modules/change_loss.py
View File

@ -22,7 +22,7 @@ class ChangeLoss(nn.Module):
compare_pose.to(device=device, dtype=dtype) compare_pose.to(device=device, dtype=dtype)
) )
self.loss = nn.MSELoss(reduce="sum").to( self.loss = nn.MSELoss(reduction="sum").to(
device=device, dtype=dtype) device=device, dtype=dtype)
self.register_buffer( self.register_buffer(