You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When you specify config.dataset, this can be a tfds dataset name (like cifar10) or the path to a directory (see the Colab vit_jax_augreg.ipynb for an illustration). The implementation of the data loading from tfds / directory is in the file vit_jax/input_pipeline.py.
If you want to use the dataset from the command line, you'll also have to extend the config, in particular this place:
This repo has used cifar as the dataset for training. If I want to use my own dataset, how can I revise the code ?
The text was updated successfully, but these errors were encountered: