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

Function build_fp16_sgd

caffe2/python/optimizer.py:2232–2234  ·  view source on GitHub ↗
(model, base_learning_rate, **kwargs)

Source from the content-addressed store, hash-verified

2230
2231
2232def build_fp16_sgd(model, base_learning_rate, **kwargs):
2233 fp16_sgd_optimizer = FP16SgdOptimizer(base_learning_rate, **kwargs)
2234 return _build(model, fp16_sgd_optimizer)
2235
2236
2237def build_ftrl(model, engine="SIMD", **kwargs):

Callers

nothing calls this directly

Calls 2

FP16SgdOptimizerClass · 0.85
_buildFunction · 0.85

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…