Fixed torch_load to load on GPU device_id rather than GPU 0

xinjli / allosaurus

Allosaurus is a pretrained universal phone recognizer for more than 2000 languages

GNU General Public License v3.0

532 stars 85 forks source link

Closed ajd12342 closed 3 years ago

ajd12342 commented 3 years ago

Fix for #4, now the loading happens on the correct GPU as well.

xinjli commented 3 years ago

looks good to me, thanks for fixing it!