search

Lightning-Universe / lightning-transformers

Flexible components pairing 🤗 Transformers with :zap: Pytorch Lightning
https://lightning-transformers.readthedocs.io
Apache License 2.0
607 stars 77 forks source link

Continued ViT support #250

Closed tanmoyio closed 1 year ago

tanmoyio commented 2 years ago

🚀 Feature

Lightning-transformers is still nlp specific, I want to work on ViT support. @SeanNaren

SeanNaren commented 2 years ago

That would be so awesome! getting some CV stuff in here would be awesome :)

Seems you're up for contributing! More than happy to help get this integrated

tanmoyio commented 2 years ago

@SeanNaren can we start tasks here ? I want to integrate, segmentation, detection and other models in vision also.

SeanNaren commented 2 years ago

Oh nice! don't worry about making an issue to capture all the tasks you want to create, just go for it :)

Nexer8 commented 2 years ago

@tanmoyio, are you also planning to introduce ViT support for object detection?

tanmoyio commented 2 years ago

@Nexer8 yes, detection, segmentation and other type or vision tasks

stale[bot] commented 1 year ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.