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()
|
||||
return self
|
||||
|
||||
def info(self): # print model information
|
||||
model_info(self)
|
||||
def info(self, verbose=False): # print model information
|
||||
model_info(self, verbose)
|
||||
|
||||
|
||||
def parse_model(d, ch): # model_dict, input_channels(3)
|
||||
|
||||
Reference in new issue
Block a user