(self, model, **kwargs)
| 84 | |
| 85 | class TestFtrl(OptimizerTestBase, TestCase): |
| 86 | def build_optimizer(self, model, **kwargs): |
| 87 | self._skip_gpu = True |
| 88 | return build_ftrl( |
| 89 | model, |
| 90 | engine=None, |
| 91 | alpha=1.0, |
| 92 | beta=0.1, |
| 93 | lambda1=0.0, |
| 94 | lambda2=0.0, |
| 95 | **kwargs |
| 96 | ) |
| 97 | |
| 98 | def check_optimizer(self, optimizer): |
| 99 | self.assertFalse(optimizer.get_auxiliary_parameters().shared) |
nothing calls this directly
no test coverage detected