Swin-Transformer
Swin-Transformer copied to clipboard
Change load_pretrained function to accept models from MoBY
Changes come from the issue I was having here: https://github.com/microsoft/Swin-Transformer/issues/181 Added a check to remove encoder prefixes as done in https://github.com/SwinTransformer/Swin-Transformer-Object-Detection Added a head check to reinit if the head key isn't present
Could you review please @ancientmooner or add someone to review?