vis

Open John122-art opened this issue 1 year ago • 1 comments

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.

Jun 27 '24 14:06 John122-art

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.

Jun 30 '24 15:06 MasterIzumi