修改模型参数 batch-size

dfl001 · 发表于 2020-11-15 16:04:27

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

复用model，最最常见的问题就是OOM了。

老大，出错了怎么办？
bs改小一点！

怎么改呀？在哪里改呀？打开模型就开始训练了呀。。。。

这是萌新一定会遇到问题，今天借平台给大家分享一下针对该问题的经验。

1、大神们经常说的bs是指batch-size ，一次有多少图像被发送到神经网络训练，越大视频整体效果越好。一般在开始设置的时候大家都会注意到，其实它是可以随时修改的，根据设备性能自己设置。（根据自己电脑硬件显存大小调节。一般用4、8、16比较多，这个数值影响训练的效率，越大收敛越快，因为在重炼时可以更改，所以显卡性能不好时可以调低点，未来换卡再调高即可，但建议不要低于4）

2、修改bs的正确步骤：
→点击train saehd.bat启动model训练
→选择model，选择显卡后，出现“两秒内按下Enter键…” 的时候，按下回车，就可以重新设置参数。
→你不想改的直接回车，到Batch_size(?:help skip:0)要设置的时候，直接在Batch_size这一行输入你想设置的数值，回车就好了。
→然后一路回车到底。