DeepID1 icon indicating copy to clipboard operation
DeepID1 copied to clipboard

关于数据集大小对测试中checkpoint恢复的影响?

Open weihualiuhupituzi opened this issue 6 years ago • 0 comments

您好,很感谢您的代码。我有一个问题想请教,由于我的机器运行vec.py代码时候内存不够大,因此导致了无法将所有的训练数据保存到dataset.pkl中,因此我适当的减小了训练数据集后将vec.py运行通过。之后直接利用30000.ckpt进行预测,可是在saver.restore(sess, '/home/weihua/git/tensorflow/faceID/DeepID1-master/checkpoint/30000.ckpt')这一步总是报错,请问是不是因为恢复的深度模型中的数据必须是与完整的数据集匹配对应上才能进行预测呢? 其中的一部分可能有关的报错为: InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [1273] rhs shape= [1283] [[Node: save/Assign_11 = Assign[T=DT_FLOAT, _class=["loc:@loss/nn_layer/biases/Variable"], use_locking=true, validate_shape=true,...... 非常感谢

weihualiuhupituzi avatar Jun 07 '18 07:06 weihualiuhupituzi