search

arpanmangal / CovidAID

COVID-19 Detection Using Chest X-Ray
http://arxiv.org/abs/2004.09803
61 stars 38 forks source link

Add support for `pytorch>=1.0` #2

Open arpanmangal opened 4 years ago

arpanmangal commented 4 years ago

Present code requires pytorch0.3 and CUDA<=8.0. The code needs to be updated to support pytorch>1.0. The main reason for using pytorch0.3 is compatibility with trained CheXNet model.

arpanmangal commented 4 years ago

Continuing discussing from #1

@SuryaKalia FYI

arpanmangal commented 4 years ago

@bartmch Yes keeping num_batches_tracked as the default value of zero tensors is the best which could be done imo. Further the keys not only differ in a module. string, but are also renamed like one has ...conv1, while other version has ...conv.1. Print the keys and compare to see all such variations.

Mayur6999 commented 4 years ago

transfer.py code is not working plz check ASAP There is a problem loading pytorch 0.3.0 and torchvision 0.2.0 plz add support