Rishit Dagli
Rishit Dagli's Blog

Follow

Rishit Dagli's Blog

Follow
💡TensorFlow Tip, use .cache

💡TensorFlow Tip, use .cache

Rishit Dagli's photo
Rishit Dagli
·Jan 1, 2021·

1 min read

Originally posted on twitter.com

💡#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:

 
Share this