Lesson 189 - RuntimeError / Load EfficientNet_B0 weights

[progr21]

Hello Team,

I faced the RuntimeError depicted in the attatched picture when trying to load the pretrained weights EfficientNet_B0 model. Could you please assist with this issue?

Thank you in advance!!

[mrdbourke]

Hi @progr21 ,

Looks like this is a known issue with torchvision, there's a quick fix is this issue thread here: #696

It should be fixed in a future version of torchvision.

Best,

Daniel