mmdeploy
mmdeploy copied to clipboard
Support profile with batch size > 1
Describe the feature
Motivation
Test the speed of model with batch size > 1
Additional context
https://github.com/open-mmlab/mmdeploy/blob/master/tools/profile.py only support batch size = 1. Could we support the speed test with batch size > 1 ?
@chenyiminch Hi, thanks for your attention, this feat would be added soon.
@chenyiminch Hi, could you try with https://github.com/open-mmlab/mmdeploy/pull/868 and see if there's any issue?