Skip to content

Latest commit

 

History

History
6 lines (4 loc) · 412 Bytes

File metadata and controls

6 lines (4 loc) · 412 Bytes

高效加载模型权重(Memory-efficient Model Weight Loading)

本目录包含示例代码,演示如何 更高效地加载模型权重

  • memory-efficient-state-dict.ipynb
    该笔记本展示了如何使用 PyTorch 的 load_state_dict 方法 优化模型权重加载,以减少 内存占用 并提高 加载速度