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:
		
							parent
							
								
									b80f5e6214
								
							
						
					
					
						commit
						bd24a8316d
					
				| @ -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__" | ||||||
|   } |   } | ||||||
|  | |||||||
| @ -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__" | ||||||
|   } |   } | ||||||
|  | |||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user