Hi,
it turned out the dataset class I had been using had a race condition bug. Whenever I used more than
one process to fetch the data, I stumbled with weird data.
Once fixed the bug and removed the track_running_stats=False flag, everything started to work as expected.
Best,
Luca