Incompatible `decoder.ckpt` tensors

[duncster94]

Hello,

I'm trying to load decoder.ckpt from the annotation_model_v1 file here using nn_models.Decoder.load_state. However, the tensor names are incompatible and the loading fails.

See this issue for additional details.

Any help is appreciated!