search

vivjay30 / Cone-of-Silence

The Cone of Silence:
MIT License
151 stars 22 forks source link

Update separation_by_localization.py to support inference on CPU #6

Closed abidlabs closed 3 years ago

abidlabs commented 3 years ago

Loading the current model without GPU doesn't work, since torch still tries to load the model on GPU. This fixes it.