Weight Initialization

[anais2390]

Hi everyone !

I was wondering if there is a way, using monai UNet(), to choose the initialization method of the weights ?

Thanks !

[wyli]

all the networks are extensions of torch.nn.Module, so I think you may follow this one https://stackoverflow.com/a/49433937