You could probably just wrap F.interpolate into a nn.Module ans still use it in your sequential model. Have a look at this post for an example.
1 Like
You could probably just wrap F.interpolate into a nn.Module ans still use it in your sequential model. Have a look at this post for an example.