diff options
| author |  Eddie Zhou <eddz@google.com> | 2018-09-05 15:24:38 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-05 15:42:44 -0700 |
| commit | 47b1af2a3a724a5d783ae06ca0e0e78b30e0799b (patch) | |
| tree | 6a7595d1210fbb18a2d2e6adf5fc3b47caf644bf /tensorflow/python/estimator | |
| parent | ebf6d259fd4c57114c17646e40fdcfa4a1472972 (diff) | |
Expose an axis argument for VocabInfo, which allows for warm-starting of the second axis of Tensors through tf.train.warm_start. Note that the underlying initializer already has this functionality (for example, for output layers).
PiperOrigin-RevId: 211709879
Diffstat (limited to 'tensorflow/python/estimator')
| -rw-r--r-- | tensorflow/python/estimator/estimator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/estimator/estimator.py b/tensorflow/python/estimator/estimator.py index e44a69b374..0f20acefdf 100644 --- a/tensorflow/python/estimator/estimator.py +++ b/tensorflow/python/estimator/estimator.py @@ -2056,7 +2056,7 @@ class WarmStartSettings( var_name_to_vocab_info: [Optional] Dict of variable names (strings) to `tf.estimator.VocabInfo`. The variable names should be "full" variables, not the names of the partitions. If not explicitly provided, the variable - is assumed to have no vocabulary. + is assumed to have no (changes to) vocabulary. var_name_to_prev_var_name: [Optional] Dict of variable names (strings) to name of the previously-trained variable in `ckpt_to_initialize_from`. If not explicitly provided, the name of the variable is assumed to be same |