MegEngine / swin-transformer

Swin-Transformer implementation in MegEngine. This is a showcase for training on GPU with less memory by leveraging MegEngine DTR technique.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MegEngine/swin-transformer Stargazers