diff options
| author |  ZhipengShen <szpssky@gmail.com> | 2017-04-21 12:07:43 +0800 |
|---|---|---|
| committer |  GitHub <noreply@github.com> | 2017-04-21 12:07:43 +0800 |
| commit | 5f606e4d306f8cd637ff2509326278e6096a6e6f (patch) | |
| tree | c649297b1d1082970c64ee9b120c940d1b825295 | |
| parent | 623dd8361325d22d1487b6a2ec67c54065dd726b (diff) | |
fix slim README error
vgg.vgg_16(images) will return two values.
| -rw-r--r-- | tensorflow/contrib/slim/README.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/slim/README.md b/tensorflow/contrib/slim/README.md index c8842dd57b..61148c0b26 100644 --- a/tensorflow/contrib/slim/README.md +++ b/tensorflow/contrib/slim/README.md @@ -447,7 +447,7 @@ vgg = tf.contrib.slim.nets.vgg images, labels = ... # Create the model. -predictions = vgg.vgg_16(images) +predictions, _ = vgg.vgg_16(images) # Define the loss functions and get the total loss. loss = slim.losses.softmax_cross_entropy(predictions, labels) |