Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

Freed-Wu opened this issue · comments

commented

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?