Weight Initialization

Did you load the state dict afterwards?
Could you try my approach of iterating the children?