Hi. I tried your solution. It seems that it works for a net with single module, i.e., a model with a single Sequential block. How can we register a hook for children of a module in a mutli-module net, e.g., ResNet?