You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Swin Transforms were added to TorchVision on #5491 thanks to the work of @xiaohu2015 and @jdsgomes. At that point, the official code of SwinV2 was not released so we decided to postpone its addition. Later @YosuaMichael has refactored the class to make it compatible with 3d use-cases at #6088. With the code of V2 now available, we should consider adding it to TorchVision.
Here is what needs to be done:
Examine if it's possible to extend the existing Swin Transformer class to support both V1 and V2 models. An example of such an extension was recently done for EfficientNets. On the other hand, if the implementations are substantially different, we can follow the approach we took for MobileNets v2 and v3 and use separate classes.
🚀 The feature
Swin Transforms were added to TorchVision on #5491 thanks to the work of @xiaohu2015 and @jdsgomes. At that point, the official code of SwinV2 was not released so we decided to postpone its addition. Later @YosuaMichael has refactored the class to make it compatible with 3d use-cases at #6088. With the code of V2 now available, we should consider adding it to TorchVision.
Here is what needs to be done:
Motivation, pitch
We should keep TorchVision fresh by offering the latest influential architectures to support the work of ML researchers and practitioners.
Alternatives
No response
Additional context
No response
cc @datumbox
The text was updated successfully, but these errors were encountered: