Error loading pre-trained models

[nathanb97]

Hello,

When I load the models in the demo / inference.py code and with the associated yaml (inference.yaml) I have the following layers which don't load (I think all layers):

_IncompatibleKeys(missing_keys=['final_layer.weight', 'final_layer.bias'], unexpected_keys=['stage4.2.fuse_layers.1.0.0.0.weight', 'stage4.2.fuse_layers.1.0.0.1.weight', ....