Add saving_listeners to tf.estimator.TrainSpec so users may pass tf.estimator.CheckpointSaverListeners.

PiperOrigin-RevId: 302665659
Change-Id: Ia28cf8c5d1da87a55f47606ee0e12f5ebc1f27f7
This commit is contained in:
A. Unique TensorFlower 2020-03-24 08:03:39 -07:00 committed by TensorFlower Gardener
parent b80f5e6214
commit bd24a8316d
2 changed files with 8 additions and 0 deletions

View File

@ -15,6 +15,10 @@ tf_class {
name: "max_steps" name: "max_steps"
mtype: "<type \'property\'>" mtype: "<type \'property\'>"
} }
member {
name: "saving_listeners"
mtype: "<type \'property\'>"
}
member_method { member_method {
name: "__init__" name: "__init__"
} }

View File

@ -15,6 +15,10 @@ tf_class {
name: "max_steps" name: "max_steps"
mtype: "<type \'property\'>" mtype: "<type \'property\'>"
} }
member {
name: "saving_listeners"
mtype: "<type \'property\'>"
}
member_method { member_method {
name: "__init__" name: "__init__"
} }