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
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
