Got it. I’m looking at the graph the tensorflow code and it seems that only gamma and beta are going into the optimizer.
VBN is used in the Discriminator after each convolution and before the non-linearities.
Let me know if you need more info about how VBN is used in my code.

Screenshot (9).png1566×489 42.9 KB