ERROR: Assign requires shapes of both tensors to match. lhs shape= [1001] rhs shape= [10]
利用tensorflow/slim将训练的cifar10分类网络输出为带权重的模型,报错:
ERROR: Assign requires shapes of both tensors to match. lhs shape= [1001] rhs shape= [10]
solution:
slim默认分类网络为ImageNet,因此类别数为1000+1(背景),如果用cifar10训练,需...
原创
2018-12-26 15:05:36 ·
2262 阅读 ·
0 评论