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 https://github.com/DemisEom/SpecAugment/pull/31 - https://github.com/seriousran/SpecAugment/commit/b1d218fcb8117639a66b86a75472bda12097e3c3 2. Update sparse_image_warp_pytorch.py https://github.com/DemisEom/SpecAugment/pull/33 - https://github.com/seriousran/SpecAugment/commit/1e5b59297ff27ddf0a63bc1cf0d7c308a65ab5a7 3. ImportError: Cannot load backend TkAgg requires tk - https://github.com/seriousran/SpecAugment/commit/39f849d279f2b06d1cfc2b8deb99452c7892e081

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.