(
model,
base_learning_rate,
parameters=None,
max_gradient_norm=None,
allow_lr_injection=False,
**kwargs
)
| 2284 | |
| 2285 | |
| 2286 | def build_storm( |
| 2287 | model, |
| 2288 | base_learning_rate, |
| 2289 | parameters=None, |
| 2290 | max_gradient_norm=None, |
| 2291 | allow_lr_injection=False, |
| 2292 | **kwargs |
| 2293 | ): |
| 2294 | storm_optimizer = StormOptimizer(lr=base_learning_rate, **kwargs) |
| 2295 | return _build( |
| 2296 | model, |
| 2297 | storm_optimizer, |
| 2298 | max_gradient_norm=max_gradient_norm, |
| 2299 | allow_lr_injection=allow_lr_injection, |
| 2300 | ) |
| 2301 | |
| 2302 | |
| 2303 | def build_adadelta( |
searching dependent graphs…