This needs double-checking but I believe when sampling across different splits the epoch_step still increments forward (not starting from 0 when moving from train to val), which isn't necessarily problematic (but maybe it is) and it needs documenting.
This needs double-checking but I believe when sampling across different splits the
epoch_stepstill increments forward (not starting from 0 when moving from train to val), which isn't necessarily problematic (but maybe it is) and it needs documenting.