diff --git a/explaining_framework/metric/base.py b/explaining_framework/metric/base.py index fc18ed5..d55250c 100644 --- a/explaining_framework/metric/base.py +++ b/explaining_framework/metric/base.py @@ -39,7 +39,6 @@ class Metric(ABC): **kwargs (optional): Additional keyword arguments passed to the model. """ - print(args, kwargs) with torch.no_grad(): out = self.model(*args, **kwargs)[0] diff --git a/explaining_framework/utils/explanation/adjust.py b/explaining_framework/utils/explanation/adjust.py index 1aa639f..726e65e 100644 --- a/explaining_framework/utils/explanation/adjust.py +++ b/explaining_framework/utils/explanation/adjust.py @@ -64,7 +64,5 @@ class Adjust(object): return mask def absolute(self, mask: FloatTensor) -> FloatTensor: - print("######################### MASK") - print(mask) mask_ = torch.abs(mask) return mask_ diff --git a/explaining_framework/utils/explanation/io.py b/explaining_framework/utils/explanation/io.py index 6a31567..ab66334 100644 --- a/explaining_framework/utils/explanation/io.py +++ b/explaining_framework/utils/explanation/io.py @@ -26,7 +26,6 @@ def explanation_verification(exp: Explanation) -> bool: def save_explanation(exp: Explanation, path: str) -> None: data = copy.copy(exp).to_dict() for k, v in data.items(): - print(k, v) if isinstance(v, torch.Tensor): data[k] = v.detach().cpu().tolist()