Skip to content

  • 体验新版
    • 正在加载...
  • 登录
  • PaddlePaddle
  • Paddle
  • 合并请求
  • !27298

P
Paddle
  • 项目概览

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
  • 代码
    • 文件
    • 提交
    • 分支
    • Tags
    • 贡献者
    • 分支图
    • Diff
  • Issue 1423
    • 列表
    • 看板
    • 标记
    • 里程碑
  • 合并请求 543
  • Wiki 0
    • Wiki
  • 分析
    • 仓库
    • DevOps
  • 项目成员
  • Pages
P
Paddle
  • 项目概览
    • 项目概览
    • 详情
    • 发布
  • 仓库
    • 仓库
    • 文件
    • 提交
    • 分支
    • 标签
    • 贡献者
    • 分支图
    • 比较
  • Issue 1,423
    • Issue 1,423
    • 列表
    • 看板
    • 标记
    • 里程碑
  • 合并请求 543
    • 合并请求 543
  • Pages
  • 分析
    • 分析
    • 仓库分析
    • DevOps
  • Wiki 0
    • Wiki
  • 成员
    • 成员
  • 收起侧边栏
  • 动态
  • 分支图
  • 创建新Issue
  • 提交
  • Issue看板

Support load state_dict from save_params/persistables !27298

  • Report abuse
!27298 已合并 9月 14, 2020 由 saxon_zh@saxon_zh 创建
#<User:0x00007f7e1fe9d370>
  • 概览 0
  • 提交 4
  • 变更 2

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暂时不支持很好地报错提示
指派人
分配到
审核者
Request review from
无
里程碑
无
分配里程碑
工时统计
标识: paddlepaddle/Paddle!27298
Source branch: github/fork/chenwhql/saveload/load_state_dict_from_save_params
渝ICP备2023009037号

京公网安备11010502055752号

网络110报警服务 Powered by GitLab CE v13.7
开源知识
Git 入门 Pro Git 电子书 在线学 Git
Markdown 基础入门 IT 技术知识开源图谱
帮助
使用手册 反馈建议 博客
《GitCode 隐私声明》 《GitCode 服务条款》 关于GitCode
Powered by GitLab CE v13.7