let weight_init = ref 1.0