eugene.models.init_weights

eugene.models.init_weights(model, initializer='kaiming_normal', **kwargs)

Initialize the weights of a model.

Parameters:

  • model (LightningModule) – The model to initialize.

  • initializer (str, optional) – The name of the initializer to use, by default “kaiming_normal”

  • **kwargs – Additional arguments to pass to the initializer.

Return type:

None