Hi,
I think the problem happens before you try to convert it to onnx no? It seems to happen when you do load_state_dict()?
And the problem appears to be that the structure of the saved weights is not the same as the struture of the ConvLSTM that you now have.