如何搭建神经网络,神经网络入门必备!
目录
废话少说,第一步,搭建网络,标注好图片
注意问题1:
data_loader
data_loader 用pytorch自带的,shuffle为True,shuffle为False会出现不收敛的情况。
关于图片数量
如果图片量比较大,先不要上大数据,先用小规模数据,比如200张训练。
如果网络收敛正常,测试训练集精度很高,可以加大图片数据规模,继续训练。
关于训练:
loss如果出现不断下降变小,再测试一下训练集,如果预测准确,说明网络搭建训练环境ok,否则网络搭建训练环境中有错误,先排查错误。
然后再加大图片量,继续训练,loss同样需要不断减小;如果很快就不减少了,说明网络欠拟合,或者网络设计不合理,或者loss函数不合理,或者网络选择的不合适。
然后再测试验证集,
训练集精度比验证集高