Use timm feature forward
Closed this issue · 3 comments
AlexanderNevarko commented
Use timm feature forward for more architectures compatibility.
timm docs
Be aware for transformer models
viacheslavm21 commented
Backbone extraction is changed for models available in timm.
All other code is left unchanged.
Added a readme guide how to choose different backbone types.
viacheslavm21 commented
Tested with convnext, resnet, vit and unicom
viacheslavm21 commented
Done