def METHOD_NAME(self, trainer, pl_module) -> None:
    self.sparsified = deepcopy(pl_module.model).eval()
    self.data_sparsifier = self.data_sparsifier_class(**self.data_sparsifier_args)
    _attach_model_to_data_sparsifier(self.sparsified, self.data_sparsifier)
    self.data_sparsifier.step()
    self.data_sparsifier.squash_mask()  # currently squashes params for all mask
    _log_sparsified_level(self.sparsified, self.data_sparsifier)