New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add warp transducer as submodule #5

Closed

vincentqb wants to merge 1 commit into master from warp-transducer

Owner

vincentqb commented Dec 17, 2020 •

edited

Loading

This pull request introduces rnnt_loss and RNNTLoss in torchaudio.transducer using HawkAaron's warp-transducer.

The python interface remains currently the same as the original.
Load replace pybind by torchbind warp-transducer#1 as a submodule. Similar to comment, we could instead use git subtree since we update the code to replace the original bindings.
Update cmdclass to use custom clean here.
Add a unit test for both CPU and GPU.

Environment variables need to be set before installing, as mentioned here and colab notebook.

export CUDA_HOME=...
export CUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME
export LD_LIBRARY_PATH="$CUDA_HOME/extras/CUPTI/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH=$CUDA_HOME/lib64:$LIBRARY_PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
export CFLAGS="-I$CUDA_HOME/include $CFLAGS"
python setup.py install

vincentqb force-pushed the warp-transducer branch 2 times, most recently from f4afbbb to 3640ba9 Compare

December 17, 2020 02:57


          add warp transducer as submodule

vincentqb force-pushed the warp-transducer branch from 3640ba9 to 6451657 Compare

December 17, 2020 03:46

Owner Author

vincentqb commented Dec 17, 2020

Replaced by pytorch#1099

vincentqb closed this

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet