tensorflow / tpu

Reference models and tools for Cloud TPUs.
https://cloud.google.com/tpu/
Apache License 2.0
5.21k stars 1.77k forks source link

Availability of Pretrained EfficientNets Checkpoints #377

Open xhluca opened 5 years ago

xhluca commented 5 years ago

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 5 years ago

+1

chuong98 commented 5 years ago

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 5 years ago

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.

mosheliv commented 5 years ago

Thank you!!!!

On Mon, Jun 17, 2019 at 4:18 PM Mingxing Tan notifications@github.com wrote:

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.

— You are receiving this because you commented. Reply to this email directly, view it on GitHub https://github.com/tensorflow/tpu/issues/377?email_source=notifications&email_token=AC7IWCYVEW7ZSNNYVJPT4A3P24GATA5CNFSM4HRF7ZWKYY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGODXZ7XGI#issuecomment-502528921, or mute the thread https://github.com/notifications/unsubscribe-auth/AC7IWCYGU2U43YM2P6P7RZ3P24GATANCNFSM4HRF7ZWA .

digital-thinking commented 5 years ago

Anyone has weights for the B7 version?

HarshitSheoran commented 5 years ago

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

jebibault commented 5 years ago

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 3 years ago

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 https://github.com/tensorflow/tpu/issues/652