Commit 8bca3e4e authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

tf.contrib.checkpoint.NumpyState for saving/restoring NumPy arrays with TF checkpoints 

A bit of extra infrastructure in checkpointable restore (save was already done) to support Python callbacks.

The same strategy should work for any Python state, although it's confined to non-pickled NumPy arrays at the moment.

PiperOrigin-RevId: 209085928
parent 4d5f6fb8
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment