您的位置:首页 > 理论基础 > 计算机网络

theano下CNN网络验证测试自己的数据集

2017-08-08 20:25 253 查看
本文仅用于记录在修改过程中遇到的问题,随时更新。

我最近在做修改LeNet5网络,本来是测试mnist的数据集,打算测试一下自己的。测试mnist时,已经可以把最佳模型保存并挑测试数据进行测试打印结果。

在修改的过程中遇到的一些问题,记录下来。

1. 运行CNN.py后,程序报错

ValueError: Shape mismatch: x has 64 cols(and 10 rows) but  y has 784 rows (and 10 cols)

原因:load_data 里面改了读取文件,但是程序里没有改n_in和n_out。

2. 训练次数两次就停了

batch_size的大小不合适,同时学习速率等参数都要改

3. 自己封装的数据集检查是否按照train valid test三组

4. 隐藏层和逻辑回归函数修改后都能跑通,主程序卡在了第一层卷积层,还在找问题所在

 

未完待续。。。。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: