diff options
| author |  Derek Murray <mrry@google.com> | 2016-07-01 13:34:17 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-07-01 14:48:13 -0700 |
| commit | f17b2ff5c79da8f79fd6c3776952dcc78727d7b7 (patch) | |
| tree | 68226c159a004fe00fcd4641396e8bce41e9c5ba /tensorflow/examples/image_retraining | |
| parent | f8bd3aa8f239df106054d70443ed39081cef7917 (diff) | |
Validate the argument to `tf.name_scope()` and fix bug in op name validation.
Change: 126447618
Diffstat (limited to 'tensorflow/examples/image_retraining')
| -rw-r--r-- | tensorflow/examples/image_retraining/retrain_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/examples/image_retraining/retrain_test.py b/tensorflow/examples/image_retraining/retrain_test.py index 837d5af8b8..91108abde0 100644 --- a/tensorflow/examples/image_retraining/retrain_test.py +++ b/tensorflow/examples/image_retraining/retrain_test.py @@ -70,9 +70,9 @@ class ImageRetrainingTest(test_util.TensorFlowTestCase): def testAddFinalTrainingOps(self): with tf.Graph().as_default(): with tf.Session() as sess: - bottleneck = tf.placeholder(tf.float32, - [1, retrain.BOTTLENECK_TENSOR_SIZE], - name=retrain.BOTTLENECK_TENSOR_NAME) + bottleneck = tf.placeholder( + tf.float32, [1, retrain.BOTTLENECK_TENSOR_SIZE], + name=retrain.BOTTLENECK_TENSOR_NAME.split(':')[0]) retrain.add_final_training_ops(5, 'final', bottleneck) self.assertIsNotNone(sess.graph.get_tensor_by_name('final:0')) |