Congratulations on achieving such excellent results. I have modified the code you provided by incorporating my ideas. However, during visualization and validation, I encountered an issue with "Missing key(s) in state_dict:". I presume the state dict was predefined by you, leading to this inconsistency. Therefore, I would like to ask where in the code the state dict is defined, what exactly is written into it, or how you view this issue. Thank you.
Hi,
Do you add more layers to the network or modify the name of sub-modules? If so, the loading process cannot find the corresponding key of the weights, and the error will be raised.
Hello,
Congratulations on achieving such excellent results. I have modified the code you provided by incorporating my ideas. However, during visualization and validation, I encountered an issue with "Missing key(s) in state_dict:". I presume the state dict was predefined by you, leading to this inconsistency. Therefore, I would like to ask where in the code the state dict is defined, what exactly is written into it, or how you view this issue. Thank you.