loop efficiency w/ cupy

[prasad-aditya]

Related Issue

Fixes N/A

Summary

• I have read CONTRIBUTING.md to understand how to contribute to this repository :)

<Please summarize what you are trying to achieve, what changes you made, and how they acheive the desired result.>
Added cupy as a dependency, rewrote loop accordingly.

Previous runtime: 0.14577968645095826s
New runtime: 0.006239746809005737s

ISSUES

Local machine does not have CUDA installed, so cannot rely on it to pass unit testing or install cupy.

Unit Tests

If your changes touch the audio module, please run all of the audio tests and paste the output here. Likewise for image, text, & video. If your changes could affect behavior in multiple modules, please run the tests for all potentially affected modules. If you are unsure of which modules might be affected by your changes, please just run all the unit tests.

Audio

python -m unittest discover -s augly/tests/audio_tests/ -p "*"

Image

python -m unittest discover -s augly/tests/image_tests/ -p "*_test.py"
# Or `python -m unittest discover -s augly/tests/image_tests/ -p "*.py"` to run pytorch test too (must install `torchvision` to run)

Text

python -m unittest discover -s augly/tests/text_tests/ -p "*"

Video

python -m unittest discover -s augly/tests/video_tests/ -p "*"

All

python -m unittest discover -s augly/tests/ -p "*"

Other testing

If applicable, test your changes and paste the output here. For example, if your changes affect the requirements/installation, then test installing augly in a fresh conda env, then make sure you are able to import augly & run the unit test

[prasad-aditya]

test_python check not passing due to absence of CUDA.

[prasad-aditya]

Can possibly use this action in workflow? @zpapakipos @jbitton

[zpapakipos]

Overall this looks great! Since you're adding a new optional dependency, we should document that so users know how to add gpu support when installing augly. Can you add a note about this in the audio README "Installation" section? For example you could add after what's already in that section, "If you have CUDA in your environment & want to use GPU-acceleration to speed up the audio augmentations, you can install augly using the following command: pip install augly[av,gpu]."