Right now I’m trying to replicated the input dimension size of a loaded model to convert the content into ONNX. Initially I was given an error for not having enough input arguments for any people who remember my previous post (Missing key(s) in state_dict 17). I fixed my previous error and now facing another one.

partONe.png1110×488 60.8 KB

I think it has something to do with my forward function handling the data type.