ConditionalBN2d

class torchelie.nn.ConditionalBN2d(channels, cond_channels, momentum=0.8)
condition(z)
forward(x, z=None)
training: bool