search

insitro / ChannelViT

Channel Vision Transformers: An Image Is Worth C x 16 x 16 Words
https://arxiv.org/abs/2309.16108
Other
47 stars 6 forks source link

adds `hubconf.py` for loading model weights using torch hub #7

Closed srinivasans-insitro closed 6 months ago

srinivasans-insitro commented 6 months ago

adds hubconf.py to load model weights using torch hub and updates README with description of pre-trained model weights and API to load them