https://github.com/AISyLab/AISY_Framework/blob/3782c45e41cba22262e877551b55ba99f44c4f45/custom/custom_datasets/datasets.py#L7 I assume this should be 50_000 not 500_000.