added different weight init per layer