spark-deep-learning icon indicating copy to clipboard operation
spark-deep-learning copied to clipboard

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

Open yobdoy opened this issue 3 years ago • 1 comments

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

to

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

yobdoy avatar Apr 07 '22 16:04 yobdoy

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

AlessiOrl avatar Jun 02 '22 11:06 AlessiOrl