MCPcopy Create free account
hub / github.com/pytorch/pytorch / build_gftrl

Function build_gftrl

caffe2/python/optimizer.py:2245–2249  ·  view source on GitHub ↗
(model, engine="", **kwargs)

Source from the content-addressed store, hash-verified

2243
2244
2245def build_gftrl(model, engine="", **kwargs):
2246 if engine == "SIMD":
2247 assert core.IsOperator("GFtrl_ENGINE_SIMD")
2248 gftrl_optimizer = GFtrlOptimizer(engine=engine, **kwargs)
2249 return _build(model, gftrl_optimizer)
2250
2251
2252def build_adagrad(

Callers 1

build_optimizerMethod · 0.90

Calls 2

GFtrlOptimizerClass · 0.85
_buildFunction · 0.85

Tested by 1

build_optimizerMethod · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…