This line of code seems to work now:
print((model.features.module[4].conv.weight == state_dict[‘features.module.4.conv.weight’]).all())
> tensor(True, device=‘cuda:0’)

while it seemed to create the AttributeError before, which is strange.
Anyway, good to see it’s working now.