facebookresearch/metaseq

Remove apex dependency after PyTorch 2.0?

オープン

#617 opened on 2023/01/24

 (3 件のコメント) (0 件のリアクション) (1 人の担当者)Python (701 件のフォーク)batch import
better-enggood first issue

Repository metrics

Stars
 (6,195 個のスター)
PR merge metrics
 (PR metrics pending)

説明

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.

コントリビューターガイド