facebookresearch/metaseq

Remove apex dependency after PyTorch 2.0?

Open

#617 aperta il 24 gen 2023

Vedi su GitHub
 (3 commenti) (0 reazioni) (1 assegnatario)Python (701 fork)batch import
better-enggood first issue

Metriche repository

Star
 (6195 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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.

Guida contributor