Suppose each of my batches are (data, target, modified_variable) where modified_variable has shape (batch_size, *) (each datapoint has an associated mdofied_variable). At a given epoch, I’d like to modify this extra variable, and save it to be accessed at the following epoch, when the same batch is loaded. How can this be done?
As an example: modified_variable could be a perturbed version of data.
Thanks!