Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.
MIT License
4.9k stars 528 forks source link

[discuss] How about set profile's verbose's default value to False? #191

Closed Freed-Wu closed 1 year ago

Freed-Wu commented 2 years ago

Now,

def profile(
    model: nn.Module,
    inputs,
    custom_ops=None,
    verbose=True,
    ret_layer_info=False,
    report_missing=False,
):

How about set verbose = False to reduce some unnecessary output?