Merge pull request #13 from emedvedev/prediction-name

emedvedev · web-flow · commit 2a48d822f0b6 · 2017-10-05T22:44:05.000+02:00
Give a name to the prediction op
diff --git a/aocr/model/model.py b/aocr/model/model.py
@@ -202,7 +202,8 @@ def __init__(self,
                 self.prediction = tf.cond(
                     tf.equal(tf.shape(trans_output)[0], 1),
                     lambda: trans_output[0],
-                    lambda: trans_output
+                    lambda: trans_output,
+                    name='prediction',
                 )
 
             if not self.forward_only:  # train

Original file line number	Diff line number	Diff line change
`@@ -202,7 +202,8 @@ def __init__(self,`
`202`	`202`	`self.prediction = tf.cond(`
`203`	`203`	`tf.equal(tf.shape(trans_output)[0], 1),`
`204`	`204`	`lambda: trans_output[0],`
`205`		`- lambda: trans_output`
	`205`	`+ lambda: trans_output,`
	`206`	`+ name='prediction',`
`206`	`207`	`)`
`207`	`208`
`208`	`209`	`if not self.forward_only: # train`