PyTorch: simple way to launch, train, use PyTorch models on any device
The accelerate module was created for PyTorch users who like to write the training loop of PyTorch models but are reluctant to write and maintain the boilerplate code needed to use multi-GPUs/TPU/fp16.
$
pkg install py311-accelerateOrigin
misc/py-accelerate
Size
4.69MiB
License
APACHE20
Maintainer
yuri@FreeBSD.org
Dependencies
8 packages
Required by
2 packages