Increasing number of workers did not help. I will stick with your second solution. By the way, could you give me an insight on why this would not occur in the next epochs, but just during the first epoch?