tensorflow / tpu

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

Does efficientnet support TF2? #994

Open tfausten opened 1 year ago

tfausten commented 1 year ago

The code in models/official/efficientnet/ in the r2.10 branch contain imports of tensorflow.contrib, which is not available in TF2.

https://github.com/tensorflow/tpu/blob/30ed88e2103bdb203334898d94a59d09987d7e85/models/official/efficientnet/autoaugment.py#L27

Does the efficientnet code support training using a TF2 runtime?