https://github.com/SwinTransformer/Feature-Distillation/issues/32