Open saichandax opened 4 months ago
To implement Swin_s transformer model, model is segregated into submodules where each module is dependent on the other. The sequence of submodules are
Currently working on ShiftedWindowAttention submodule. Task Updates regarding ShiftedWindowAttention:
For your reference, added torch and referenc graphs of ShiftedWindowAttention submodule. Torch graph:
Reference graph:
reference model is from: https://github.com/pytorch/vision/blob/main/torchvision/models/swin_transformer.py