1、加载全部数据到显存中然后训练
这种方式对于小数据量数据没问题,一次加注全部数据不会爆内存
2、一个batch加载,加载一个完毕后训练再加载反反复复
这种方式无论多少数据,只要设置batch合适都不会爆内存


深度学习训练两种读取和训练方式差异
最新推荐文章于 2025-09-28 13:51:58 发布
本文探讨了两种有效的内存管理方法:一次性加载所有数据适用于小数据集;批量加载则适用于大数据集,通过合理设置batch大小,避免内存溢出,确保训练过程顺利进行。
929

被折叠的 条评论
为什么被折叠?



