Swin Transformers

Provides an implementation of Swin Transformers, attempting to adhere to best practices for reproducibility

Creating Environment

$ conda env create -f environment.yml
$ conda activate swin

Running the model

$ python scripts/train.py

Importing the model into other code

pip install -e .

from swin import build_swin_model

model = build_swin_model(out_classes=20)

Running the tests

pytest