(self, optimizer)
| 54 | return build_sgd(model, base_learning_rate=0.1, **kwargs) |
| 55 | |
| 56 | def check_optimizer(self, optimizer): |
| 57 | self.assertTrue(optimizer.get_auxiliary_parameters().shared) |
| 58 | self.assertFalse(optimizer.get_auxiliary_parameters().local) |
| 59 | for param in optimizer.get_auxiliary_parameters().shared: |
| 60 | tensor = workspace.FetchBlob(param) |
| 61 | np.testing.assert_allclose(np.array([1.0]), tensor, atol=1e-5) |
| 62 | |
| 63 | |
| 64 | class TestMultiPrecisionSgd( |
no test coverage detected