使用小数据集在 GPU 上完全训练 Pytorch Lightning

问题描述

我正在使用一个足够小的架构和一个足够小的数据集,它们可以很好地适应 GPU,而无需批量工作。

在使用 pytorch Lightning Trainer 时,如何避免我的整个数据集在每个时期都从 cpu 移动到 GPU?可以吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)