关于测试训练好的模型的问题

来源:8-20 训练流程代码

kankan123

2019-05-09

老师您好,我有这样一个问题,训练的时候使用batch来训练,大小是80,但测试的时候我希望对单独的一张图片进行测试,这样placeholder就对应不上了,这个问题怎么解决呢
写回答

1回答

正十七

2019-05-17

你可以在测试的时候,重新建一个graph,然后载入模型,在这个重建的graph上,batch_size设为1就可以了。我们在之前的章节里有载入模型的代码。

0
1
不安分的coder
要是老师能上传一份代码就更好啦,哈哈
2020-04-15
共1条回复

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程