(
model,
base_learning_rate,
parameters=None,
max_gradient_norm=None,
allow_lr_injection=False,
**kwargs
)
| 2250 | |
| 2251 | |
| 2252 | def build_adagrad( |
| 2253 | model, |
| 2254 | base_learning_rate, |
| 2255 | parameters=None, |
| 2256 | max_gradient_norm=None, |
| 2257 | allow_lr_injection=False, |
| 2258 | **kwargs |
| 2259 | ): |
| 2260 | adagrad_optimizer = AdagradOptimizer(alpha=base_learning_rate, **kwargs) |
| 2261 | return _build( |
| 2262 | model, |
| 2263 | adagrad_optimizer, |
| 2264 | max_gradient_norm=max_gradient_norm, |
| 2265 | allow_lr_injection=allow_lr_injection, |
| 2266 | ) |
| 2267 | |
| 2268 | |
| 2269 | def build_wngrad( |
searching dependent graphs…