From c16671fc740a71d4de653b3023ae805feb307b8e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 24 Apr 2022 12:07:05 -0700 Subject: [PATCH] Add `print(results)` override for PyTorch Hub results (#7559) --- models/common.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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):