unit_test / data /checkpoints.md
herrius's picture
Upload 259 files
32b542e

Checkpoints

We provide links for you to download the pre-trained weights of some Uni-Perceiver models.

ModelLinkParamsHidden sizeIntermediate sizeNum. of headsEnc layers
Uni Perceiverbase Downloadtorch Downloadds124M76830721212
Uni Perceiver MoEbaseDownloadds167M76830721212
Uni PerceiverlargeDownloadds354M102440961624
Uni Perceiver MoElargeDownloadds505M102440961624
  • Models torch are pretrained in our current codebase, while models ds are pretrained in our previous codebase with DeepSpeed engine.

  • Params in the table is the parameters required during model deployment for image-text tasks. Note that it may be different on other tasks.

  • More pre-trained weights will be released.