【发布时间】:2016-12-09 19:57:20
【问题描述】:
我有一个非常复杂的网络,它在我的 gpu 上占用了大量内存。我发现如果我训练和测试我的数据(这是标准情况),内存使用量是我只进行训练的两倍。真的有必要测试我的数据吗?或者它只是用于可视化,即显示我的网络是否过度拟合或类似的东西?
我认为这是必要的,但我不知道原因。我的问题是:如何分开训练和测试?我知道你能做到
test_initialization: false
但是,如果我想测试我的网络,我该怎么做呢?
提前致谢!
【问题讨论】:
-
您的图层中还有 TEST 阶段吗?
-
不,应该吗? @卡西
-
如果你这样做了,那么你可以使用命令行测试。如果没有,你可以写一个脚本。我正在为你写一个答案。
标签: deep-learning caffe conv-neural-network