code: def initialize_module(self):
          
          kwargs = self._get_params_for('module')
          module = self.module
          is_initialized = isinstance(module, torch.nn.Module)
  
          if kwargs or not is_initialized:
              if is_initialized:
                  module = type(module)
  
              if (is_initialized or self.initialized_) and self.verbose:
                  msg = self._format_reinit_msg("module", kwargs)
                  print(msg)
  
              module = module(**kwargs)
  
          self.module_ = module.to(self.device)
          return self
docs: Initializes the module.
  
          Note that if the module has learned parameters, those will be
          reset.