Commit
·
d0f98c0
1
Parent(s):
b985122
Update yolo.py with model_info(verbose=False)
Browse files- 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)
|