pytorch 避免在ML编辑-编译-运行循环中重新加载权重/数据集

nzkunb0c  于 2022-11-09  发布在  其他
关注(0)|答案(1)|浏览(105)

在机器学习中,编辑-编译-运行循环非常慢,因为脚本必须加载大型模型和数据集。
在过去,我通过只加载数据的一个小子集,并且在设置训练代码时不使用预先初始化的权重来避免这种情况。

3okqufwl

3okqufwl1#

使用Jupyter笔记本电脑或Google colab。
您可以一次编辑和编译一个单元格,另一个单元格中的数据集和训练权重将被保留。
不知怎的,直到刚才这才合上。

相关问题