FakeData uses the argument num_classes to return the dummy targets.
I’m not sure, why you would need a training and test dataset, as your model won’t learn anything useful and FakeData is usually used for debugging/testing.