Update yolo.py with model_info(verbose=False)
This commit is contained in:
1 parent
b9851222bf
commit
d0f98c0095
1 file changed
+2
-2
+2
-2
@@ -168,8 +168,8 @@ class Model(nn.Module):
|
|||||||
self.info()
|
self.info()
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def info(self): # print model information
|
def info(self, verbose=False): # print model information
|
||||||
model_info(self)
|
model_info(self, verbose)
|
||||||
|
|
||||||
|
|
||||||
def parse_model(d, ch): # model_dict, input_channels(3)
|
def parse_model(d, ch): # model_dict, input_channels(3)
|
||||||
|
|||||||
Reference in new issue
Block a user