feat: add a new config

SwinEncoder pos_drop fix and additional config options

Update train.py

added output scores code to model.py

Introducing the scores to the outputs

_

Bff/token optimisation

Fix timm version

feat: update pl version supports

Dependency update