Expose tf.contrib.checkpoint.PythonStateWrapper.
This makes it possible to checkpoint arbitrary python state if it can be serialized to a string. Also updates NumpyState to accept np.int32, np.int64, np.float32, np.float64 types. PiperOrigin-RevId: 212879609
Loading
Please sign in to comment