Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Feature] input_shape of get_model_complexity_info() for multiple tensors #1065

Merged
merged 17 commits into from
Apr 23, 2023
Merged

[Feature] input_shape of get_model_complexity_info() for multiple tensors #1065

merged 17 commits into from
Apr 23, 2023

Commits on Apr 22, 2023

  1. input_shape of get_model_complexity_info() for multiple tensors 

    Previously, the arg `input_shape` of get_model_complexity_info() is used
to construct only one input tensor.
This PR add support of using `input_shape` to construct multiple input
tensors.

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    4d18c9b View commit details
    Browse the repository at this point in the history

  2. re-format the error message 

    Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    86ab4f6 View commit details
    Browse the repository at this point in the history

  3. use 'is_tuple_of()' to simplify the logic

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    3f2e67f View commit details
    Browse the repository at this point in the history

  4. add description and Example block to show the usages

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    67d0098 View commit details
    Browse the repository at this point in the history

  5. the arg 'inputs' accepts not only tensors

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    070b706 View commit details
    Browse the repository at this point in the history

  6. add 'tests/test_analysis/test_print_helper.py'

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    90571c3 View commit details
    Browse the repository at this point in the history

  7. fix type hint of 'tuple of tuples of ints'

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    be5cc3c View commit details
    Browse the repository at this point in the history

  8. complement test of no input (shape) is given

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    28b1d1e View commit details
    Browse the repository at this point in the history

  9. add type hint to forward()

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    ad50ac3 View commit details
    Browse the repository at this point in the history

  10. fix scalar input error for pytorch<1.9.0

    @sjiang95
    sjiang95 committed Apr 22, 2023
    Configuration menu
    Copy the full SHA
    81a9e1b View commit details
    Browse the repository at this point in the history

Commits on Apr 23, 2023

  1. print_helper: bypass mypy type check 

    Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
    @sjiang95
    sjiang95 committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    cac6f0e View commit details
    Browse the repository at this point in the history

  2. modify class name 

    Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
    @sjiang95
    sjiang95 committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    648c263 View commit details
    Browse the repository at this point in the history

  3. Update mmengine/analysis/print_helper.py 

    accept suggestions of example block

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    @sjiang95 @zhouzaida
    sjiang95 and zhouzaida committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    64f88c1 View commit details
    Browse the repository at this point in the history

  4. Update mmengine/analysis/print_helper.py 

    accept relative import

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    @sjiang95 @zhouzaida
    sjiang95 and zhouzaida committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    8e59218 View commit details
    Browse the repository at this point in the history

  5. Update tests/test_analysis/test_print_helper.py 

    accept test unit suggestions

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    @sjiang95 @zhouzaida
    sjiang95 and zhouzaida committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    b574a56 View commit details
    Browse the repository at this point in the history

  6. ut: fix lint and typo 

    Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>
    @sjiang95
    sjiang95 committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    b86aaee View commit details
    Browse the repository at this point in the history

  7. Update mmengine/analysis/print_helper.py

    @zhouzaida
    zhouzaida committed Apr 23, 2023
    Configuration menu
    Copy the full SHA
    b4fdfcf View commit details
    Browse the repository at this point in the history