Lightning-Universe / lightning-Covid19

Classification for covid-19 chest X-ray images using Lightning
https://pytorchlightning.github.io/lightning-Covid19
MIT License
56 stars 18 forks source link

Error in running Experiments. #21

Open amitbcp opened 4 years ago

amitbcp commented 4 years ago
  1. The _parse_batch() function throws a Key Error while trying to fetch x["PA"] . This is not consistent with torchxrayvision
  2. The Kornia based data augmentation adds an additional dimension while doing unsqueeze.