How I can load the tf hub weights in this repo?

Question

How I can load the tf hub weights in this repo?

vkmavani opened this issue 4 years ago · 5 comments

https://colab.research.google.com/github/google/automl/blob/master/efficientnetv2/tfhub.ipynb

'efficientnetv2-b0-21k-ft1k': f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-b0-21k-ft1k/{hub_type}',
'efficientnetv2-b1-21k-ft1k': f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-b1-21k-ft1k/{hub_type}',
'efficientnetv2-b2-21k-ft1k': f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-b2-21k-ft1k/{hub_type}',
'efficientnetv2-b3-21k-ft1k': f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-b3-21k-ft1k/{hub_type}',
'efficientnetv2-s-21k-ft1k':  f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-s-21k-ft1k/{hub_type}',
'efficientnetv2-m-21k-ft1k':  f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-m-21k-ft1k/{hub_type}',
'efficientnetv2-l-21k-ft1k':  f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-l-21k-ft1k/{hub_type}',
'efficientnetv2-xl-21k-ft1k':  f'gs://cloud-tpu-checkpoints/efficientnet/v2/hub/efficientnetv2-xl-21k-ft1k/{hub_type}',

leondgarse commented 4 years ago

Done

Answer 1 · 2021-07-24T07:21:00.000Z

Cool, they have all these types! I'll try if I can load these...

Answer 2 · 2021-07-24T14:10:54.000Z

I'm updating the new converted model weights, but takes time. You can also do it yourself refer to my converting part.

Answer 3 · 2021-07-25T02:43:01.000Z

Thank you very much for quick reply, the main reason not using that models from TF-hub is that I cannot access the layers of the model.

Again Thanks for your work.

Answer 4 · 2021-07-25T06:21:34.000Z

Ya, that's why I create this. The official implementation sometimes not so convenient.