Michael Benayoun d4e4efce68 Initial support for symbolic tracing with torch.fx allowing dynamic axes (#13579) ...

* Symbolic trace dynamic axes support for BERT like models (albert, bert, distilbert, mobilebert, electra, megatron-bert)
* Sanity checks before tracing that make sure the model to trace is supported
* Adapted to PyTorch 1.9

Co-authored-by: Michael Benayoun <michael@huggingface.co>

2021-10-05 14:19:47 +02:00

13 KiB

Raw Blame History

View Raw