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.
Try it out for yourself: