Nystromformer icon indicating copy to clipboard operation
Nystromformer copied to clipboard

LRA cifar10.py never stop

Open liranringel opened this issue 2 years ago • 0 comments

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?

liranringel avatar Dec 25 '22 17:12 liranringel