From 3e40ff67ad0dabfe90ff63678dce6b4943b03d0c Mon Sep 17 00:00:00 2001
From: Wlad <9556979+gosticks@users.noreply.github.com>
Date: Wed, 17 Feb 2021 20:53:39 +0100
Subject: [PATCH] fix minor runtime error

---
 config.yaml            | 2 +-
 modules/change_loss.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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(