search

lukemelas / EfficientNet-PyTorch

A PyTorch implementation of EfficientNet
Apache License 2.0
7.81k stars 1.52k forks source link

convert tensorflowjs #109

Closed kewin1807 closed 4 years ago

kewin1807 commented 4 years ago

i have custom train data with efficientNet. How can i convert the model to tensorflow js

lukemelas commented 4 years ago

For anyone with this question in the future, look at the TensorFlow version of EfficientNet: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet