Add transformers to models - Githubissues

initze / thaw-slump-segmentation

MIT License

9 stars 3 forks source link

Add transformers to models #114

Open initze opened 2 months ago

initze commented 2 months ago

Problem

we have currently a lot of models implemented from segmentation models and timm
transformers (ViT, Swin) are not implemented yet

Potential solution

timm has some implementations of transformers
- https://github.com/huggingface/pytorch-image-models/tree/main/timm/models
huggingface might be cool as well