How to export detectron2.modeling.backbone.swin.SwinTransformerBlock by torch.jit.script?

[XZRRRRRR]

I find it's not easy to export detectron2.modeling.backbone.swin.SwinTransformerBlock by torch.jit.script. Cause in the init function where self.H and W is defined to None; but in forward function them are used as H, W = self.H, self.W. I notice that these two things are redefined in other module, but it can not suit for using torch.jit.script to export it. What should I do??