2 篇文章 含有標籤「PyTorch」

手算的 LayerNorm 數值不對？

2024年8月20日 · 閱讀時間約 4 分鐘

Z. Yuan

Maintainer

閒來無事，動手算算。恩，怎麼不對啊？

PyTorch 的 List 陷阱

2024年2月20日 · 閱讀時間約 2 分鐘

Z. Yuan

Maintainer

發現與解決 PyTorch OOM 問題。