DEV Community

loading...

💡TensorFlow Tip, use .cache

rishitdagli profile image Rishit Dagli Originally published at twitter.com ・1 min read

💡#TensorFlowTip

Use .cache to save on some ops like opening file and reading data during each epoch

  • transformations before cache are only run for 1st epoch
  • can cache in-memory or on-disk
  • not repeatedly perform expensive CPU ops

See the speedup with .cache in this image.

Alt Text

Try it out for yourself:

Discussion (0)

pic
Editor guide