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

Method build_optimizer

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

Source from the content-addressed store, hash-verified

263
264class TestSmartDecayAdam(OptimizerTestBase, LRModificationTestBase, TestCase):
265 def build_optimizer(self, model, **kwargs):
266 self._skip_gpu = False
267 kwargs['beta1'] = 0.0
268 return build_adam(model, base_learning_rate=0.1, use_smart_decay=True, **kwargs)
269
270 def check_optimizer(self, optimizer):
271 self.assertTrue(optimizer.get_auxiliary_parameters().shared)

Callers

nothing calls this directly

Calls 1

build_adamFunction · 0.90

Tested by

no test coverage detected