deepspeed自动生成导出脚本zero_to_fp32.py

脚本传参：

参数1 deepspeed保存pth的目录

参数2 导出文件的路径

# 实例
python zero_to_fp32.py ./ ./torch_model.pth
© 允许规范转载
如果觉得我的文章对你有用，请随意赞赏