Merge pull request clovaai#95 from varshaneya/master

removed align_corners=True
tungth07 · Oct 23, 2019 · 9a3a141 · 9a3a141
2 parents 9a6f667 + 53509ea
commit 9a3a141
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/transformation.py b/modules/transformation.py
@@ -30,7 +30,7 @@ def forward(self, batch_I):
         batch_C_prime = self.LocalizationNetwork(batch_I)  # batch_size x K x 2
         build_P_prime = self.GridGenerator.build_P_prime(batch_C_prime)  # batch_size x n (= I_r_width x I_r_height) x 2
         build_P_prime_reshape = build_P_prime.reshape([build_P_prime.size(0), self.I_r_size[0], self.I_r_size[1], 2])
-        batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border', align_corners=True)
+        batch_I_r = F.grid_sample(batch_I, build_P_prime_reshape, padding_mode='border')
 
         return batch_I_r