File size: 473 Bytes
6ed4a9c |
1 2 3 4 5 6 7 8 9 10 11 |
# Copyright (c) OpenMMLab. All rights reserved.
from .complexity_analysis import (ActivationAnalyzer, FlopAnalyzer,
activation_count, flop_count,
parameter_count, parameter_count_table)
from .print_helper import get_model_complexity_info
__all__ = [
'FlopAnalyzer', 'ActivationAnalyzer', 'flop_count', 'activation_count',
'parameter_count', 'parameter_count_table', 'get_model_complexity_info'
]
|