赞
踩
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
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)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。