mirror of
https://github.com/hzwer/ECCV2022-RIFE.git
synced 2025-12-16 16:37:51 +01:00
Update IFNet_HDv2.py
This commit is contained in:
@@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user