Add cuda detection and fix problem due to PyTorch Bug #43227 - Githubissues

ksanjeevan / crnn-audio-classification

UrbanSound classification using Convolutional Recurrent Networks in PyTorch

MIT License

380 stars 80 forks source link

Add cuda detection and fix problem due to PyTorch Bug #43227 #19

Open javadba opened 3 years ago

javadba commented 3 years ago

Reference: https://github.com/pytorch/pytorch/issues/43227