/Swin-Transformer

https://github.dev/kuro7766/Swin-Transformer

Primary LanguagePythonMIT LicenseMIT

paper link: https://arxiv.org/pdf/2103.14030.pdf

requirements:

pip install timm yacs amp
pip install --ignore-installed PyYAML

git clone https://github.com/NVIDIA/apex
cd apex
pip3 install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
cd ..