当前位置:   article > 正文

调试AlexNet问题_alexnet中 acc上不去怎么办

alexnet中 acc上不去怎么办

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)

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/546193
推荐阅读
  

闽ICP备14008679号