Skip to content

3 12 月, 2023

[已解決][PyTorch] 編譯(compile)後模型權重多出 "_orig_mod" 的前綴問題

Last Updated on 2023-12-04 by Clay

問題描述

在 2023 年初,PyTorch 的 2.0 版本新增了一個 torch.compile() 的新功能,讓我們能夠在模型訓練/推理時能夠進一步提昇速度。與混合精度訓練的協同工作,經常能使我的訓練速度提昇一倍左右。

Read More »[已解決][PyTorch] 編譯(compile)後模型權重多出 "_orig_mod" 的前綴問題