Fixing inplace operation error (#37)

* Update decoding.py * Update encoding.py
fepegar · Jul 30, 2023 · caace5c · caace5c
1 parent 6d3f74b
commit caace5c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/unet/decoding.py b/unet/decoding.py
@@ -133,7 +133,7 @@ def forward(self, skip_connection, x):
             connection = self.conv_residual(x)
             x = self.conv1(x)
             x = self.conv2(x)
-            x += connection
+            x = x + connection
         else:
             x = self.conv1(x)
             x = self.conv2(x)

diff --git a/unet/encoding.py b/unet/encoding.py
@@ -143,7 +143,7 @@ def forward(self, x):
             connection = self.conv_residual(x)
             x = self.conv1(x)
             x = self.conv2(x)
-            x += connection
+            x = x + connection
         else:
             x = self.conv1(x)
             x = self.conv2(x)