add init weights