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