text
text copied to clipboard
Revamp Pytorch Benchmark implementations for pytorch_struct and Attention is All You Need models
🚀 Feature
Revamp Pytorch Benchmark implementations for pytorch_struct and Attention is All You Need models
Motivation Since we deprecated pytorch.legacy, we need to revamp these tutorials to make them adhere to newer TorchText APIs.
Where can these be found? I'd be happy to take a stab at this.
This?
https://github.com/pytorch/benchmark/blob/121ea0230d9f675928625b9a9c83d4cd9772f27f/torchbenchmark/models/pytorch_struct/init.py
https://github.com/pytorch/benchmark/tree/121ea0230d9f675928625b9a9c83d4cd9772f27f/torchbenchmark/models/attention_is_all_you_need_pytorch
Just started this. Tracking here.
- [ ] pytorch_struct
- [ ] aiayn
I think the pytorch_struct refactor is ready for review: https://github.com/pytorch/benchmark/pull/673 cc @mthrok @abhinavarora
Due to a recent code change in torchtext(https://github.com/pytorch/text/pull/1654), another workaround is added: https://github.com/pytorch/benchmark/pull/795