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

Clean up on the model architectures test.

Browse Source 
PiperOrigin-RevId: 293891433
Change-Id: Ia4caf5cc793feb6f6f0d697368717000020990eb
This commit is contained in:
Yanhui Liang 2020-02-07 14:08:39 -08:00 committed by TensorFlower Gardener
parent e05fdf7a7a
commit 74463d793c
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/keras/tests/model_architectures_test.py
View File

@ -81,16 +81,10 @@ class TestModelArchitectures(keras_parameterized.TestCase):
model_architectures.MySubclassModel,
}
custom_objects.update(subclass_custom_objects)
elif ('subclassed' in model_fn.__name__ and
save_format in ['h5', 'hdf5', 'keras']):
elif ('subclassed' in model_fn.__name__ and save_format == 'h5'):
self.skipTest('Saving the model to HDF5 format requires the model to be '
'a Functional model or a Sequential model.')
# TODO(b/147493902): Remove this skipTest once fixed.
if ('stacked_rnn' in model_fn.__name__
and save_format in ['h5', 'hdf5', 'keras']):
self.skipTest('Stacked RNN model is not compatible with h5 save format.')
saved_model_dir = self._save_model_dir()
model_data = model_fn()
model = model_data.model