mlpen / Nystromformer

Apache License 2.0
356 stars 41 forks source link

LRA cifar10.py never stop #17

Open liranringel opened 1 year ago

liranringel commented 1 year ago

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?