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

Method build_optimizer

caffe2/python/optimizer_test.py:161–165  ·  view source on GitHub ↗
(self, model, **kwargs)

Source from the content-addressed store, hash-verified

159
160class TestRowWiseAdagrad(OptimizerTestBase, TestCase):
161 def build_optimizer(self, model, **kwargs):
162 self._skip_gpu = True
163 return build_adagrad(
164 model, base_learning_rate=1.0, lars=0.5, rowWise=True, **kwargs
165 )
166
167 def check_optimizer(self, optimizer):
168 self.assertFalse(optimizer.get_auxiliary_parameters().shared)

Callers

nothing calls this directly

Calls 1

build_adagradFunction · 0.90

Tested by

no test coverage detected