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))

[Carl1998-CY]

Is your problem solved? I'm having the same issue too

[kiansierra]

No but I believe this PR Would fix it #21