Update vit_jax.ipynb - Githubissues

google-research / vision_transformer

Apache License 2.0

10.48k stars 1.3k forks source link

Update vit_jax.ipynb #265

Closed limesqueezy closed 1 year ago

limesqueezy commented 1 year ago

When tfds.builder("cifar10").info is run (through input_pipeline.get_dataset_info(dataset, 'train')['num_classes']) a dictionary is returned which does has an empty split attribute. I suppose this was not the case with the tensorflow-datasets version 6 months ago neither was it when tensorflow-datasets==4.8.2, as mentioned here. A minor workaround to make go past this trivial error is to change the order of the cells.

google-cla[bot] commented 1 year ago

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.