From 53509eab0135b2c47aca3910f3ca23f894bf62ce Mon Sep 17 00:00:00 2001 From: varshaneya Date: Wed, 23 Oct 2019 11:35:37 +0530 Subject: [PATCH] removed align_corners=True Got this error in Pytorch 1.1.0: TypeError: grid_sample() got an unexpected keyword argument 'align_corners' --- modules/transformation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/transformation.py b/modules/transformation.py index ded2684..7c4dd5a 100755 --- a/modules/transformation.py +++ b/modules/transformation.py @@ -30,7 +30,7 @@ class TPS_SpatialTransformerNetwork(nn.Module): 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