调试AlexNet问题

1.from tensorflow.contrib.data import Dataset 加载不了

解决:https://blog.csdn.net/kwame211/article/details/78579035

在TensorFlow 1.3中,Dataset API是放在contrib包中的:

tf.contrib.data.Dataset
  • 而在TensorFlow 1.4中,Dataset API已经从contrib包中移除,变成了核心API的一员:
tf.data.Dataset

2.TypeError: map() got an unexpected keyword argument 'num_threads'

解决:原因在与map中的参数多了出来

原函数:

data = data.map(self._parse_function_train,num_threads=8,output_buffer_size=100*batch_size)

修改后:

data = data.map(self._parse_function_train)

 

你可能感兴趣的:(tensorflow)