2024年4月21日
Pytorch 如何使用 storage 实现参数 offload？
摘要： 在深入探讨 PyTorch 中的 Storage 类以及其在参数 offload 场景中的应用之前，让我们首先了解一下 PyTorch 和它的基础组件。PyTorch 是一个广泛使用的开源机器学习库，它不仅提供了强大的计算图功能和自动梯度计算，还允许开发者直接操作底层数据结构，这其中就包括 Stor 阅读全文
posted @ 2024-04-21 19:07 marsggbo 阅读(88) 评论(0) 推荐(0) 编辑