thank you very much for your reply!
I suppose this suggest a direct call for the weight_init and not applying the model.apply(weight_init) technique:
something like this:
init_layers = ['conv12', 'conv2', 'conv31', 'conv4'] # layer that are allowed to be initialized
        for i in model.named_children():
            if i[0] in init_layers: # layer name
               weights_init(i)

otherwise, I don’t see how to apply your suggestion with the model.apply() technique