search

google / uis-rnn

This is the library for the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm, corresponding to the paper Fully Supervised Speaker Diarization.
https://arxiv.org/abs/1810.04719
Apache License 2.0
1.55k stars 320 forks source link

assign gpu with arguments #91

Open pthavarasa opened 1 year ago

pthavarasa commented 1 year ago

Describe the question

if i enable gpu args.enable_cuda=True by default it use 'cuda:0' but i have multiple gpu on my pc so i think we can make an argument or environment variable CUDA_VISIBLE_DEVICES to chose manually.

My background

Have I read the README.md file?

Have I searched for similar questions from closed issues?

Have I tried to find the answers in the paper Fully Supervised Speaker Diarization?

Have I tried to find the answers in the reference Speaker Diarization with LSTM?

Have I tried to find the answers in the reference Generalized End-to-End Loss for Speaker Verification?