Need to modify kdl/transformers/keras_applications to be able to use resnet50

databricks / spark-deep-learning

Deep Learning Pipelines for Apache Spark

https://databricks.github.io/spark-deep-learning

Apache License 2.0

1.99k stars 494 forks source link

Need to modify kdl/transformers/keras_applications to be able to use resnet50 #247

Open yobdoy opened 2 years ago

yobdoy commented 2 years ago

Hi,

Per this overflow question one needs to modify /home/user/.local/lib/python3.8/site-packages/sparkdl/transformers/keras_applications.py . This happened in databricks using v 10.3.

Have to change from keras.applications import inception_v3, xception, resnet50

from keras.applications import inception_v3, xception from tensorflow.keras.applications import resnet50

AlessiOrl commented 2 years ago

did you find a way to import the library? i'm having the same problem