def test_it(self, grad_scale, layer): layer2 = grad_scale(layer, 0.33) assert layer2 is layer for param in layer.get_params(trainable=True): np.testing.assert_almost_equal(param.tag.grad_scale, 0.33) for param in layer.get_params(trainable=False): assert hasattr(param.tag, 'grad_scale') is False
def test_it(self, grad_scale, layer): layer2 = grad_scale(layer, 0.33) assert layer2 is layer for param in layer.get_params(trainable=True): assert param.tag.grad_scale == 0.33 for param in layer.get_params(trainable=False): assert hasattr(param.tag, 'grad_scale') is False