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

stripAndFreezeGraph not found

Open abuvaneswari opened this issue 8 years ago • 0 comments

Trying to do hands-on exercises given in the main github page. While doing Tensorflow transformer example (with TFImageTransformer),

from sparkdl import readImages, TFImageTransformer from sparkdl.transformers import utils import tensorflow as tf

g = tf.Graph() with g.as_default(): image_arr = utils.imageInputPlaceholder() resized_images = tf.image.resize_images(image_arr, (299, 299)) # the following step is not necessary for this graph, but can be for graphs with variables, etc frozen_graph = utils.stripAndFreezeGraph(g.as_graph_def(add_shapes=True), tf.Session(graph=g), [resized_images])

Got the following error: Traceback (most recent call last): File "", line 4, in AttributeError: 'module' object has no attribute 'stripAndFreezeGraph'

Checked transformers/utils.py code and found that there is no function stripAndFreezeGraph.

pyspark invoked with the locally assembled jar file : spark-deep-learning-assembly-0.1.0-spark2.1.jar

abuvaneswari avatar Oct 07 '17 01:10 abuvaneswari