the main Idea that I want to apply parameter initialization to specific layers. the condition will be tested by the layer name.
experts recommend model.apply(init_fn)
this iterate through modules of the model by calling children() which will call named_chidren() method
named_children output name and module children however output only the module. at this level we lose the name information.
the condition to apply the initialization cannot not be tested
Help!