I tried this but the running_mean{i} in the buffer does not get updated for some reason when I update the self.running_means[i]. In fact, I registered self.running_means[i] into the buffer directly to make sure the buffer points to the dictionary element directly, but that didn’t help either.