Solved 3 issues/PRs for pytorch. And I'm going to maintenance it instead.

[seriousran]

https://github.com/seriousran/SpecAugment

1.Fixed typo in import for pytorch #31
- seriousran@b1d218f
2. Update sparse_image_warp_pytorch.py #33
- seriousran@1e5b592
3. ImportError: Cannot load backend TkAgg requires tk
- seriousran@39f849d

pip install git+https://github.com/seriousran/SpecAugment.git

I'm going to maintenance it instead.
I will make sure that there is no problem using it in PyTorch, and TF is available in the 2.x version.
Please PR freely in my fork.