iVAE.module.weight_init

iVAE.module.weight_init(m)

Initialize weights for linear layers using Xavier normal initialization.

Parameters:

m (nn.Module) – The module to initialize. If it’s a Linear layer, weights are initialized with Xavier normal and biases are set to 0.01.