glenn-jocher commited on
Commit
c16671f
·
unverified ·
1 Parent(s): be67572

Add `print(results)` override for PyTorch Hub results (#7559)

Browse files
Files changed (1) hide show
  1. models/common.py +5 -1
models/common.py CHANGED
@@ -683,7 +683,11 @@ class Detections:
683
  return x
684
 
685
  def __len__(self):
686
- return self.n
 
 
 
 
687
 
688
 
689
  class Classify(nn.Module):
 
683
  return x
684
 
685
  def __len__(self):
686
+ return self.n # override len(results)
687
+
688
+ def __str__(self):
689
+ self.print() # override print(results)
690
+ return ''
691
 
692
 
693
  class Classify(nn.Module):