深度学习训练两种读取和训练方式差异

分类: 文章 • 2025-05-31 08:10:22

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