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)