search

arpanmangal / CovidAID

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

torch>1.0 and cpu compatible #3

Open bartmch opened 4 years ago

bartmch commented 4 years ago
  1. Rename torch<1.0 layers to be compatible with torch>1.0 layers.
  2. Make it possible to run when torch.cuda isn't available (CPU)