Added TPU code for FTRL with multiply_linear_by_lr setting (CPU and GPU
versions already added). PiperOrigin-RevId: 305428489 Change-Id: Ib2bb5b26ba06c535b96222aa1488a018d4a8f343
This commit is contained in:
		
							parent
							
								
									9a273849b2
								
							
						
					
					
						commit
						72e0b46788
					
				@ -86,6 +86,7 @@ message FtrlParameters {
 | 
			
		||||
  float lr_power = 3;
 | 
			
		||||
  float initial_accum = 4;
 | 
			
		||||
  float initial_linear = 5;
 | 
			
		||||
  bool multiply_linear_by_lr = 6;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// The Adam optimizer does not implement hyper-parameter update; use the dynamic
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user