Nystromformer
Nystromformer copied to clipboard
LRA cifar10.py never stop
When trying to run python3 cifar10.py
to prepare the cifar10 dataset, the script gets into an infinite loop because input_pipeline.get_cifar10_datasets
returns a dataset with infinite sample (it uses the repeat
function).
What do you think I should do to fix it?