I have a custom convolution layer that support two layouts: KRSC and RSKC, RS is kernel size, K is out channel, C is input channel, used by different algorithms. Now I want to load RSKC state dict to KRSC layer, I can’t find any useful hooks to achieve this. a hook named “_register_load_state_dict_pre_hook” exists but it’s a private method and don’t have guarantee that available in torch 1.5+.