from .flops_counter import add_flops_counting_methods, flops_to_string