search

intel-analytics / analytics-zoo

Distributed Tensorflow, Keras and PyTorch on Apache Spark/Flink & Ray
https://analytics-zoo.readthedocs.io/
Apache License 2.0
16 stars 3 forks source link

import tfhub models in Analytics Zoo/BigDL #664

Open guidiandrea opened 4 years ago

guidiandrea commented 4 years ago

Hello.

Is there a way to import pretrained models from TF Hub and possibly fine tune them?

glorysdj commented 4 years ago

please refer to this example: https://github.com/intel-analytics/analytics-zoo/blob/master/apps/tfnet/image_classification_inference.ipynb

guidiandrea commented 4 years ago

Thanks, but there is no reference to TFHub in the notebook you provided. Do you have any specific hints?

yangw1234 commented 4 years ago

Hi @guidiandrea , as far as I understand, there are no differences between models created from tf hub and models created from scratch. I think you can just use tf hub as you would use it in TensorFlow and adapt your code to analytics zoo TFPark. Here is the TFPark documentation. (https://analytics-zoo.github.io/master/#ProgrammingGuide/TFPark/tensorflow/)

helenlly commented 3 years ago

@guidiandrea any more questions? if no, we'll close the issue in a week, thanks