2016-12-04 2 views
2

Je souhaite utiliser le resnet intégré dans tf-slim pour une expérience rapide. Je l'ai fait selon le README dans github:AttributeError: le module 'tensorflow.contrib.slim' n'a pas d'attribut 'nets'

import tensorflow as tf 
import tensorflow.contrib.slim as slim 

resnet = tf.contrib.slim.nets.resnet_v1 

mnist = input_data.read_data_sets('MNIST_data', one_hot=True) 
x = tf.placeholder("float", shape=[None, 784]) 
y_ = tf.placeholder("float", shape=[None, 10]) 
pred = resnet.resnet_v1_50(x) 

cross_entropy = -tf.reduce_sum(y_ * tf.log(pred)) 

mais nous avons eu une telle erreur: AttributeError: module 'tensorflow.contrib.slim' has no attribute 'nets'. J'ai déjà installé la dernière version de tensorflow-0.12.0.
Comment puis-je résoudre ce problème?

Répondre

0
importation comme celui-ci

:

from tensorflow.contrib.slim.nets import resnet_v1 

slim = tf.contrib.slim