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

Method build_optimizer

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

Source from the content-addressed store, hash-verified

107 raise unittest.SkipTest("no sparse support")
108
109 def build_optimizer(self, model, **kwargs):
110 self._skip_gpu = True
111 return build_gftrl(
112 model,
113 engine=None,
114 alpha=1.0,
115 beta=0.1,
116 lambda1=0.0,
117 lambda2=0.0,
118 **kwargs
119 )
120
121 def check_optimizer(self, optimizer):
122 self.assertFalse(optimizer.get_auxiliary_parameters().shared)

Callers

nothing calls this directly

Calls 1

build_gftrlFunction · 0.90

Tested by

no test coverage detected