vis

[John122-art]

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.

[MasterIzumi]

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.