Swin-Transformer implementation in MegEngine. This is a showcase for training on GPU with less memory by leveraging MegEngine DTR technique.
-
Updated
Oct 19, 2021 - Python
Swin-Transformer implementation in MegEngine. This is a showcase for training on GPU with less memory by leveraging MegEngine DTR technique.
Reproducibility package for deterministic regime switching and feasibility inversion in Dynamic Tensor Rematerialization (DTR).
To associate your repository with the dtr topic, visit your repo's landing page and select "manage topics."