Skip to content

Latest commit

 

History

History
31 lines (22 loc) · 2.13 KB

File metadata and controls

31 lines (22 loc) · 2.13 KB

PyTorch训练项目转换

支持将PyTorch代码及预训练模型转换为PaddlePaddle代码及预训练模型。

使用方法

第一步:转换前代码预处理

由于部分PyTorch操作是目前PaddlePaddle暂不支持的操作(例如:不支持TensorBoard、自动下载模型等),因此我们需要手动将这部分操作去除或者修改,具体可参见转换前代码预处理

第二步:转换

x2paddle --convert_torch_project --project_dir=torch_project --save_dir=paddle_project --pretrain_model=model.pth
参数 作用
--convert_torch_project 当前方式为对PyTorch Project进行转换
--project_dir PyTorch的项目路径
--save_dir 指定转换后项目的保存路径
--pretrain_model **[可选]**需要转换的预训练模型的路径(文件后缀名为“.pth”、“.pt”、“.ckpt”)或者包含预训练模型的文件夹路径,转换后的模型将将保在当前路径,后缀名为“.pdiparams”

第三步:转换后代码后处理

PaddlePaddle在使用上有部分限制(例如:自定义Dataset必须继承自paddle.io.Dataset、部分情况下DataLoader的num_worker只能为0等),用户需要手动修改代码,使代码运行,具体可参见转换后代码后处理

[注意] 转换前后相应操作可以参考转换示例

致谢

感谢aiyasin为本文档贡献issue和PR,同时也感谢jstzwjrfaded-TJUfreemustard156aasdfghrdwsonggeoyeeArlanCooperFelix-python2U-makerAlexZou14skywalk163Darki-luo在论文复现营比赛过程中通过issue积极为X2Paddle反馈使用中存在的问题!