DingXiaoH / RepVGG

RepVGG: Making VGG-style ConvNets Great Again
MIT License
3.34k stars 435 forks source link

请问如果我用RepVGG作为backbone,并且在neck中有其他的操作,比如5*5卷积,这样训练出来的模型还能做结构重参数化吗? #103

Closed AssassinJay closed 2 years ago

AssassinJay commented 2 years ago

仔细读了下example_pspnet.py的代码,发现以下代码用来判断是否合并分支

for module in model.modules():
     if hasattr(module, 'switch_to_deploy'):
          module.switch_to_deploy()