experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update documentation to use proper tf.train.CheckpointOptions.

Browse Source 
PiperOrigin-RevId: 355202310
Change-Id: Ied4328f9b71cc1ba4872a284d2eb2e0367ca1227
This commit is contained in:
Haitang Hu 2021-02-02 10:44:22 -08:00 committed by TensorFlower Gardener
parent 0df420bebd
commit 937a4232b1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/training/tracking/util.py
View File

@ -2132,7 +2132,8 @@ class Checkpoint(tracking.AutoTrackable):
# You can also pass options to read(). For example this
# runs the IO ops on the localhost:
options = tf.CheckpointOptions(experimental_io_device="/job:localhost")
options = tf.train.CheckpointOptions(
experimental_io_device="/job:localhost")
checkpoint.read(path, options=options)
```