fix the full layer weight update bug