ilu) —
Activation function


latent_channels (int, optional, defaults to 4) —
Latent space channels


norm_num_groups (int, optional, defaults to 32) —
Norm num group


sample_size (int, optional, defaults to 32) —
Sample input size


scaling_factor (float, optional, defaults to 0.18215) —
The component-wise standard deviation of the trained latent space computed using the first batch of the
training set. This is used to scale the latent space 