Improved FLOPS computation (#1398)

* Improved FLOPS computation

* update comment

models/yolo.py

@@ -192,8 +192,8 @@ class Model(nn.Module):
         copy_attr(m, self, include=('yaml', 'nc', 'hyp', 'names', 'stride'), exclude=())  # copy attributes
         return m
 
-    def info(self, verbose=False):  # print model information
-        model_info(self, verbose)
+    def info(self, verbose=False, img_size=640):  # print model information
+        model_info(self, verbose, img_size)
 
 
 def parse_model(d, ch):  # model_dict, input_channels(3)