支持将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,同时也感谢jstzwjr、faded-TJU、freemustard、156aasdfg、 hrdwsong、geoyee、ArlanCooper、 Felix-python、2U-maker 、AlexZou14、skywalk163、Darki-luo在论文复现营比赛过程中通过issue积极为X2Paddle反馈使用中存在的问题!