question about the 'model.load_state_dict(checkpoint['state_dict'])',

[siyusa]

when I run the code,the result is like this :Missing key(s) in state_dict: "module.densenet121.features.denseblock1.denselayer1.......' I have been suffering with this problem for a week ,could you please help me ,I will appreciate very much.Looking forward to hearing from you

[YakinRubaiat]

I modified the heatmap code like this, it can process the state_dict correctly,
https://github.com/YakinRubaiat/chexnet/blob/ceb92ba7e43d646e9b51bcae77e647ba40ac8b59/HeatmapGenerator.py#L42

[AtulCIS]

Hi @YakinRubaiat , I am also getting the same error and I could not access the file HeatmapGenerator.py added by you

[yuyijie1995]

I met the same question , are you solved it?

[egenc]

just change the line model.load_state_dict(modelCheckpoint['state_dict']) to model.load_state_dict(modelCheckpoint['state_dict'], False)