What I expect:

get model’s state_dict and its keys
get submodules of model through these keys
register forward hook on these modules

Why I cannot use model.children:
I have select modules’ weight_dict through key word filter and it’s hard to say what kind of module will be processed after