diff --git a/models/common.py b/models/common.py index 7eca56aaa3bd..d59ee976c9f7 100644 --- a/models/common.py +++ b/models/common.py @@ -683,7 +683,11 @@ def tolist(self): return x def __len__(self): - return self.n + return self.n # override len(results) + + def __str__(self): + self.print() # override print(results) + return '' class Classify(nn.Module):