facebookresearch/metaseq

Remove apex dependency after PyTorch 2.0?

開放

#617 建立於 2023年1月24日

 (3 則留言) (0 個反應) (1 位負責人)Python (701 個分叉)batch import
better-enggood first issue

倉庫指標

星標
 (6,195 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

This is to look into whether or not we still need to use apex for speedups if out-of-the-box PyTorch 2.0 may "just work". Will require benchmarking at a few different scales to confirm.

貢獻者指南