deel.lip.initializers module

class deel.lip.initializers.SpectralInitializer(niter_spectral=3, niter_bjorck=15, k_coef_lip=1.0, base_initializer=<keras.initializers.initializers_v2.Orthogonal object>)

Bases: keras.initializers.initializers_v2.Initializer

Initialize a kernel to be 1-lipschitz orthogonal using bjorck normalization.

Parameters

  • niter_spectral – number of iteration to do with the iterative power method

  • niter_bjorck – number of iteration to do with the bjorck algorithm

  • base_initializer – method used to generate weights before applying the orthonormalization

get_config()

Returns the configuration of the initializer as a JSON-serializable dict.

Returns

A JSON-serializable Python dict.