KeyError: 'module name can\'t contain "."'

nyukat / breast_cancer_classifier

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening

https://ieeexplore.ieee.org/document/8861376

GNU Affero General Public License v3.0

840 stars 269 forks source link

KeyError: 'module name can\'t contain "."' #25

Closed ankitdata closed 4 years ago

ankitdata commented 4 years ago

getting above ERROR during stage 3 run. after research discovered that torch vision package needs to be updated to 0.2.1 (https://github.com/pytorch/vision/pull/474) this is just an FYI

zphang commented 4 years ago

Hi @ankitdata,

Is it the opposite issue? The models are not compatible with a newer torchvision version that has more restrictions on module keys. I tested the setup on a fresh install using the existing requirements and it works fine with the currently specified versions of PyTorch and torchvision.