Common weights initialisation
RMeli opened this issue · comments
Rocco Meli commented
Define weights and biases initialisation function outside of models' __init__
and perform initialisation with nn.Module.apply()
function.
Check how biases are initialised and perform the same initialisation.