From d226968b8bd5f776b7c8fa6d66ec2a71d7302102 Mon Sep 17 00:00:00 2001 From: Yang Chen Date: Wed, 13 May 2026 10:41:03 -0700 Subject: [PATCH] Make sure proportions updates are applied immediately to TF mixtures. For dynamic proportions, we need to disable prefetching to prevent prefetched data from being returned after proportions are updated. PiperOrigin-RevId: 914942843 --- CHANGELOG.md | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index c0ba3efdc..ec08be82c 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -10,9 +10,8 @@ changes. Best viewed [here](https://google-grain.readthedocs.io/en/latest/change used to save one data copy. * Adds experimental `ShapeDtypeStructProtocol` and `ShapeDtypeStruct` to represent dataset element specs. - * Updates TfMixtureIndexSampler to support datasets with weights of 0. - * Updates TfMixtureIndexSampler so that proportions could be updated during - iteration. + * Updates TfMixtureIndexSampler to accept dynamic Tensor/Variable proportions + that could be updated during runtime. * Adds profiling of multiprocess workers when using XProf profiler. To enable, set flag `grain_enable_multiprocess_worker_profiling=true` and add `"profile_subprocesses" = True` in advanced profiler options.