Einsum optimization using opt_einsum and PyTorch FX graph rewriting
-
Updated
Mar 17, 2022 - Python
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
Tune-Mode ConvBN Blocks For Efficient Transfer Learning
Add a description, image, and links to the pytorch-fx topic page so that developers can more easily learn about it.
To associate your repository with the pytorch-fx topic, visit your repo's landing page and select "manage topics."