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

Added a warning about AutoTrackable

Browse Source
This commit is contained in:
gehring 2019-06-10 15:52:13 -04:00 committed by GitHub
parent 9480262cbb
commit cbf76901b8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/training/saver.py
View File

@ -721,6 +721,9 @@ class Saver(object):
# as keys:
saver = tf.compat.v1.train.Saver({v.op.name: v for v in [v1, v2]})
```
Note: the newer `AutoTrackable` API is not supported by `Saver`. In this
case, the `tf.train.Checkpoint` class should be used.
The optional `reshape` argument, if `True`, allows restoring a variable from
a save file where the variable had a different shape, but the same number