search

jinmang2 / DOOLY

🦕 A library that handles everything with 🤗 and supports batching to models in PORORO
https://huggingface.co/jinmang2/dooly-hub
Apache License 2.0
37 stars 3 forks source link

GPU initialize problem #18

Open jinmang2 opened 2 years ago

jinmang2 commented 2 years ago

https://github.com/jinmang2/DOOLY/blob/main/dooly/tasks/base/base.py#L309