Don't CHECK-fail when weights don't match.
Don't CHECK-fail when weights don't match.

picked_weight may be smaller than target_weight because of floating point
errors.

PiperOrigin-RevId: 321358137
Change-Id: I56c866b914834c6a732e664f767d2805fb0cee99

deepmind/reverb,deepmind/reverb,deepmind/reverb,deepmind/reverb