mirror of
https://github.com/gosticks/body-pose-animation.git
synced 2025-10-16 11:45:42 +00:00
fix minor runtime error
This commit is contained in:
parent
6da9a47495
commit
3e40ff67ad
@ -52,7 +52,7 @@ pose:
|
||||
vposerPath: "./vposer_v1_0"
|
||||
temporal:
|
||||
enabled: true
|
||||
iterations: 75
|
||||
iterations: 50
|
||||
lr: 0.02
|
||||
preview:
|
||||
enable: true,
|
||||
|
||||
@ -22,7 +22,7 @@ class ChangeLoss(nn.Module):
|
||||
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)
|
||||
|
||||
self.register_buffer(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user