glenn-jocher commited on
Commit
d0f98c0
1 Parent(s): b985122

Update yolo.py with model_info(verbose=False)

Browse files
Files changed (1) hide show
  1. models/yolo.py +2 -2
models/yolo.py CHANGED
@@ -168,8 +168,8 @@ class Model(nn.Module):
168
  self.info()
169
  return self
170
 
171
- def info(self): # print model information
172
- model_info(self)
173
 
174
 
175
  def parse_model(d, ch): # model_dict, input_channels(3)
 
168
  self.info()
169
  return self
170
 
171
+ def info(self, verbose=False): # print model information
172
+ model_info(self, verbose)
173
 
174
 
175
  def parse_model(d, ch): # model_dict, input_channels(3)