experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

added test

Browse Source
This commit is contained in:
Dominic Jack 2020-06-16 09:06:15 +10:00
parent 17b7e16913
commit 57eccc7bc2
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/python/keras/engine/training_test.py
View File

@ -3328,6 +3328,16 @@ class TestTrainingWithMetrics(keras_parameterized.TestCase):
outer_model.fit(np.ones((10, 1)), np.ones((10, 1)), batch_size=10) outer_model.fit(np.ones((10, 1)), np.ones((10, 1)), batch_size=10)
self.assertEqual([m.name for m in outer_model.metrics], self.assertEqual([m.name for m in outer_model.metrics],
['loss', 'acc2', 'mean', 'mean1', 'mean2']) ['loss', 'acc2', 'mean', 'mean1', 'mean2'])
def test_subclassed_model_with_empty_list_attr(self):
class ModelSubclass(training_module.Model):
def __init__(self):
self.empty_list = []
inputs = layers_module.Input(shape=())
outputs = inputs + 1
super(ModelSubclass, self).__init__(inputs, outputs)
ModelSubclass() # empty_list attr assignment should not raise
class BareUpdateLayer(layers_module.Layer): class BareUpdateLayer(layers_module.Layer):