本目录包含示例代码,演示如何 更高效地加载模型权重。
- memory-efficient-state-dict.ipynb:
该笔记本展示了如何使用 PyTorch 的load_state_dict
方法 优化模型权重加载,以减少 内存占用 并提高 加载速度。
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||
本目录包含示例代码,演示如何 更高效地加载模型权重。
load_state_dict
方法 优化模型权重加载,以减少 内存占用 并提高 加载速度。