Update transformation.py

tungth07 · Jan 21, 2020 · 1344512 · 1344512
1 parent 78d7ed9
commit 1344512
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/transformation.py b/modules/transformation.py
@@ -32,7 +32,7 @@ def forward(self, batch_I):
         build_P_prime_reshape = build_P_prime.reshape([build_P_prime.size(0), self.I_r_size[0], self.I_r_size[1], 2])
 
         if torch.__version__ > "1.2.0":
-            batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border', align_corner=True)
+            batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border', align_corners=True)
         else:
             batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border')