(
model,
base_learning_rate,
max_gradient_norm=None,
allow_lr_injection=False,
**kwargs
)
| 2318 | |
| 2319 | |
| 2320 | def build_adam( |
| 2321 | model, |
| 2322 | base_learning_rate, |
| 2323 | max_gradient_norm=None, |
| 2324 | allow_lr_injection=False, |
| 2325 | **kwargs |
| 2326 | ): |
| 2327 | adam_optimizer = AdamOptimizer(alpha=base_learning_rate, **kwargs) |
| 2328 | return _build( |
| 2329 | model, |
| 2330 | adam_optimizer, |
| 2331 | max_gradient_norm=max_gradient_norm, |
| 2332 | allow_lr_injection=allow_lr_injection, |
| 2333 | ) |
| 2334 | |
| 2335 | def build_decay_adagrad( |
| 2336 | model, |
searching dependent graphs…