关于repconv switch-to-deploy的问题

[pierrelzw]

RepVGG/repvgg.py

Line 106 in 5c2e359

assert isinstance(branch, nn.BatchNorm2d)

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