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