def __call__(self, module):

    def init(m):
        if self.wholemodule:
            uniform_init(m, self.a, self.b, self.bias)
        else:
            layername = m.__class__.__name__
            basesname = _get_bases_name(m)
            if len((set(self.layer) & set(([layername] + basesname)))):
                uniform_init(m, self.a, self.b, self.bias)
    module.apply(init)
    if hasattr(module, '_params_init_info'):
        update_init_info(module, init_info=self._get_init_info())
def __call__(self, module):

    def init(m):
        if self.wholemodule:
            uniform_init(m, self.a, self.b, self.bias)
        else:
            layername = m.__class__.__name__
            basesname = _get_bases_name(m)
            if len((set(self.layer) & set(([layername] + basesname)))):
                uniform_init(m, self.a, self.b, self.bias)
    module.apply(init)
    if hasattr(module, '_params_init_info'):
        update_init_info(module, init_info=self._get_init_info())