Model initialisation

[sahar90o]

Hi everyone,

I am trying to build an ensemble model for 3D segmentation in MONAI.
to do so I want to train the same model (for example Unet) with different weights. Is there any way to initialize the model weights with different methods using normal, Xavier, Kaimimg, ...?
Thank you
Zahra

[KumoLiu]

Hi @sahar90o, we have some utilities:

MONAI/monai/networks/utils.py

Line 303 in b1c8b42

def normal_init(

And you can also refer to:

MONAI/monai/networks/blocks/localnet_block.py

Line 286 in 08d5728

if initializer == "kaiming_uniform":

Hope it helps, thanks.