Skip to content

Commit

Permalink
added logging info to cleaner
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianGroeger96 committed Apr 9, 2024
1 parent 464b9f1 commit c6906ac
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/cleaner/selfclean_cleaner.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import numpy as np
import scienceplots # noqa: F401
import sklearn # noqa: F401
from loguru import logger
from torch.utils.data import Dataset
from tqdm.auto import tqdm

Expand Down Expand Up @@ -94,6 +95,7 @@ def fit(
self.class_labels = class_labels
self.N, self.D = emb_space.shape
self.condensed_size = int(self.N * ((self.N - 1) / 2))
logger.info(f"Fitting cleaner on representation space: {emb_space.shape}")

if self.memmap:
dist_file = self.memmap_path / "dist_matrix.dat"
Expand Down

0 comments on commit c6906ac

Please sign in to comment.