diff --git a/config.yaml b/config.yaml index a756e60..4aa0528 100644 --- a/config.yaml +++ b/config.yaml @@ -52,7 +52,7 @@ pose: vposerPath: "./vposer_v1_0" temporal: enabled: true - iterations: 75 + iterations: 50 lr: 0.02 preview: enable: true, diff --git a/modules/change_loss.py b/modules/change_loss.py index aace423..f1733bd 100644 --- a/modules/change_loss.py +++ b/modules/change_loss.py @@ -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(