michalwols / accelerate

A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

michalwols/accelerate Stargazers