tensorflow / cloud

The TensorFlow Cloud repository provides APIs that will allow to easily go from debugging and training your Keras and TensorFlow code in a local environment to distributed training in the cloud.
https://github.com/tensorflow/cloud
Apache License 2.0
377 stars 91 forks source link

Performance issues in src/python/tensorflow_cloud/core/tests/testdata/keras_tuner_cifar_example.py(P2) #369

Open DLPerf opened 3 years ago

DLPerf commented 3 years ago

Hello,I found a performance issue in src/python/tensorflow_cloud/core/tests/testdata/keras_tuner_cifar_example.py , train_dataset = train_dataset.map was called without num_parallel_calls. I think it will increase the efficiency of your program if you add this.

The same issues also exist in test_dataset = test_dataset.map

Here is the documemtation of tensorflow to support this thing.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.