Closed PedroCarrinhoRibeiro closed 2 years ago
Thanks for pointing out the problem. It was because the code only handled batch_normalize=1
case (2
was not handled correctly).
The problem has been fixed in the latest code (https://github.com/jkjung-avt/tensorrt_demos/commit/f53b5ae9b004489463a407d8e9b230f39230d051) now.
Hi,
So in Darknet, it's possible to turn on Cross-iteration Batch Normalization by changing batch_normalize=1 to batch_normalize=2 in the convolutional layers in the .cfg file.
However, when I tried to convert a model that used CBN from Darknet to trt, the model couldn't make any inference.
Is CBN implemented in your repo? If so, what changes do I need to make to use it?
Thanks in advance.
Edit: P.S. The model works completely fine in trt if I use the regular batch normalization instead.