Skip to content

Commit 36a232b

Browse files
committed
Fix #169, remove recompute_scale_factor arg from interpolate.
1 parent 5b1c60f commit 36a232b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

effdet/efficientdet.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,8 +129,7 @@ def __init__(self,
129129
self.align_corners = None if mode == 'nearest' else align_corners
130130

131131
def forward(self, input: torch.Tensor) -> torch.Tensor:
132-
return F.interpolate(
133-
input, self.size, self.scale_factor, self.mode, self.align_corners, recompute_scale_factor=True)
132+
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
134133

135134

136135
class ResampleFeatureMap(nn.Sequential):

0 commit comments

Comments
 (0)