PyTorch: Package for applying ao techniques to GPU models
torchao is analogous to a great API in Keras to view the visualization of the model which is very helpful while debugging your network. The aim is to provide information complementary to, what is not provided by print(your_model) in PyTorch.
$
pkg install py311-torchaoOrigin
misc/py-torchao
Size
15.0MiB
License
BSD3CLAUSE
Maintainer
yuri@FreeBSD.org
Dependencies
3 packages
Required by
0 packages