DingXiaoH / RepVGG

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

关于repconv switch-to-deploy的问题 #93

Open pierrelzw opened 1 year ago

pierrelzw commented 1 year ago

https://github.com/DingXiaoH/RepVGG/blob/5c2e359a144726b9d14cba1e455bf540eaa54afc/repvgg.py#L106

使用这个 assert 的原因是? 我基于repConv block 搭建了网络,并尝试在 doCheckpoint 时 export onnx 模型,所以我会在每次保存模型时把模型转换为deploy模型。 基于单卡训练的时候,一起正常。但是使用多机多卡,就触发了这个assert,所以像问问这个 assert 的作用