fix minor runtime error

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

View File

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

View File

@ -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(