Removing debugging printing

This commit is contained in:
araison 2023-01-02 23:39:46 +01:00
parent 9ad5adb33e
commit 68449ad678
3 changed files with 0 additions and 4 deletions

View File

@ -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]

View File

@ -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_

View File

@ -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()