Created by: chenwhql
PR types
New features
PR changes
APIs
Describe
支持使用paddle.load接口从save_params/save_persistables载入state_dict
前提:所给目录下找不到的*.pdparams
并且找不到*.pdopt
, 并且找不到静态图的模型文件__model__
- 分散存储为多个参数文件:
- 从目录读取文件名列表后载入
- 如果存在非参数文件,load时报错(除非手动添加文件,一般不会有)
- 从目录读取文件名列表后载入
- 存储为一个参数文件:
- 暂不支持,建议使用io.load_program_state
- 因为这种存储丢失了变量信息和实际tensor的映射,用户需要在动态图下构建这些信息比较复杂,不如在静态图下使用
- 这种情况下,也能够正常载入,但仅会载入文件的第一个tensor,由于load op支持seek配置载入数据的位置,默认文件可以不被完全使用,这种case暂时不支持很好地报错提示
- 暂不支持,建议使用io.load_program_state