fan_small_12_p16_224_se_attn Forward Issue

[kiansierra]

Hi there I receive the following error when running the forward method on models with FANBlock_SE TypeError: forward() missing 2 required positional arguments: 'H' and 'W'

An example of code that reproduces the issue

from FAN.models.fan import fan_small_12_p16_224_se_attn
import torch

model = fan_small_12_p16_224_se_attn()
model(torch.ones(2,3,224,224))