An observation: DataGenerator\'s getitem() must return a batch. This means that if you don\'t make getitem() parallel under the hood then i