efficientnet-jax icon indicating copy to clipboard operation
efficientnet-jax copied to clipboard

Published 20 hours ago • verified publisher icon rwightman

Sample code for Loading Pretrained Model in JAX?

Open riven314 opened this issue 3 years ago • 0 comments

Thanks for the great work! I have a few questions:

  1. I saw you released a collections of pre-trained model checkpoints, do you have any sample code on how to load them in JAX?
  2. Does this pretrained models have comparable performance against that from PT/ TF?
  3. Also you mentioned the model weights are ported from timm and tensorflow, I am curious to know how did you do the porting? (i.e. conversion from PT/ TF model weights to JAX's)

riven314 avatar Feb 02 '22 17:02 riven314