search

smivv / pytorch-django-chexnet

CheXNet implementation for Classification and Localization of Thoracic Diseases with Django server
16 stars 7 forks source link

Possible to run on Pytorch with CPU support only? #2

Closed taewookim closed 6 years ago

taewookim commented 6 years ago

what modifications would i need to make?

smivv commented 6 years ago

@taewookim I've tried to run it on cpu, but it does not work, because model was trained in gpu.

taewookim commented 6 years ago

really? is that some kind of pytorch limitation? pretty sure in TF, you can train on GPU and run on CPU no problem...

smivv commented 6 years ago

@taewookim I tried with no luck, of course you are welcome to pull request with your implementation.

taewookim commented 6 years ago

thanks @smivv . will poke around and see if i find anything

smivv commented 6 years ago

@taewookim let me know if you need something.