lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reformat code

Browse Source
This commit is contained in:
Francois Chollet 2023-04-26 15:31:30 -07:00
parent de4ef7bddc
commit d03973f1da
1 changed files with 5 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
keras_core/layers/layer_test.py
View File

@ -140,20 +140,12 @@ class LayerTest(testing.TestCase):
return x
layer = ActivityRegularizer(activity_regularizer="l1")
layer(
np.ones(
1,
)
)
layer(np.ones((1,)))
self.assertLen(layer.losses, 1)
self.assertAllClose(layer.losses[0], 0.01)
# losses are reset upon call
layer(
np.ones(
1,
)
)
layer(np.ones((1,)))
self.assertLen(layer.losses, 1)
self.assertAllClose(layer.losses[0], 0.01)
@ -164,30 +156,18 @@ class LayerTest(testing.TestCase):
return x
layer = LossLayer()
layer(
np.ones(
1,
)
)
layer(np.ones((1,)))
self.assertLen(layer.losses, 1)
self.assertAllClose(layer.losses[0], 1.0)
# losses are reset upon call
layer = LossLayer()
layer(
np.ones(
1,
)
)
layer(np.ones((1,)))
self.assertLen(layer.losses, 1)
self.assertAllClose(layer.losses[0], 1.0)
# It works inside a model
model = models.Sequential([layer])
model(
np.ones(
1,
)
)
model(np.ones((1,)))
self.assertLen(model.losses, 1)
self.assertAllClose(model.losses[0], 1.0)