search

open-mmlab / mmrazor

OpenMMLab Model Compression Toolbox and Benchmark.
https://mmrazor.readthedocs.io/en/latest/
Apache License 2.0
1.48k stars 229 forks source link

[Bug] Can't use RepVGG with AutoSlim #97

Open twmht opened 2 years ago

twmht commented 2 years ago

It's very similar to #79, it's a bug of the parser, except the exception is that it reached maximum recursion when calling trace_bn_conv_links (https://github.com/open-mmlab/mmrazor/blob/master/mmrazor/models/pruners/structure_pruning.py#L124).

To reproduce, simply replace the model (https://github.com/open-mmlab/mmrazor/blob/master/configs/pruning/autoslim/autoslim_mbv2_supernet_8xb256_in1k.py) with RepVGG.

I am refactoring the parser to make it more general and easily to debug. Stay tuned.

pppppM commented 2 years ago

Sorry for taking so long to get back to you. Next, I will focus on MMRazor, and solve the related issues you raised as soon as possible.