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