tensorflow/tpu

Availability of Pretrained EfficientNets Checkpoints

xhluca opened this issue · 8 comments

xhluca commented

With the release of EfficientNet, it is interesting great to see the partial release of checkpoints:

We have provided a list of EfficientNet checkpoints for EfficientNet-B0, EfficientNet-B1, EfficientNet-B2, and EfficientNet-B3.

Is there any reason why the weights for B4 to B7 are not released yet? Is there plans to release them in the future?

Link to directory: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet

Thanks!

mosheliv commented

+1

chuong98 commented

Can you also provide the checkpoints at the couple first epochs during training b0-b3. We would like to try to apply Lottery Ticket Hypothesis on EfficientNet b0-b3. Many thanks

mingxingtan commented

I have released B4 and B5:

https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/efficientnet-b4.tar.gz (ImageNet top1 acc=82.6%)
https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/efficientnet-b5.tar.gz (ImageNet top1 acc=83.2%)

We will release B6 and B7 in future.

  • 👍16
  • 👀1
mosheliv commented
digital-thinking commented

Anyone has weights for the B7 version?

  • 👀2
HarshitSheoran commented

@digital-thinking Not yet available, I am also waiting.

jebibault commented

The B6 and B7 checkpoints are available: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet
Do you know if you intend to convert them to weights and make them available?
Thank you very much!

bnascimento commented

Can someone clarify how to use these checkpoints to fine tune to a custom dataset?
I tried to put them on the output model_dir so they can resume from there, but keep stumbling into missing or changed operations? See #652