Update IFNet_HDv2.py

This commit is contained in:
hzwer
2021-02-02 18:09:52 +08:00
committed by GitHub
parent bf790c2ffd
commit 0d33876da2

View File

@@ -48,7 +48,7 @@ class IFBlock(nn.Module):
flow = x flow = x
if self.scale != 1: if self.scale != 1:
flow = F.interpolate(flow, scale_factor=self.scale, mode="bilinear", flow = F.interpolate(flow, scale_factor=self.scale, mode="bilinear",
align_corners=False) * self.scale align_corners=False)
return flow return flow