muhd-umer / pvt-flax Goto Github PK
View Code? Open in Web Editor NEWUnofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions.
Home Page: https://arxiv.org/abs/2102.12122
License: MIT License