Fixed dataset path in train tests in examples#5092

vfdev-5 wants to merge 1 commit intomaingoogle/flax:mainfrom

fix-dataset-path-examplesgoogle/flax:fix-dataset-path-examplesCopy head branch name to clipboard