def test_load_definition(self):
K.set_image_dim_ordering("th")
kmodel, input_data, output_data = TestModels.kmodel_seq_lenet_mnist()
keras_model_json_path, keras_model_hdf5_path = self._dump_keras(kmodel, dump_weights=True)
bmodel = DefinitionLoader.from_json_path(keras_model_json_path)
WeightLoader.load_weights_from_kmodel(bmodel, kmodel)
self.assert_allclose(bmodel.forward(input_data), kmodel.predict(input_data))
def test_load_definition(self):
K.set_image_dim_ordering("th")
kmodel, input_data, output_data = TestModels.kmodel_seq_lenet_mnist()
keras_model_json_path, keras_model_hdf5_path = dump_keras(kmodel, dump_weights=True)
bmodel = DefinitionLoader.from_json_path(keras_model_json_path)
WeightLoader.load_weights_from_kmodel(bmodel, kmodel)
self.assert_allclose(bmodel.forward(input_data), kmodel.predict(input_data))
def compare_model(self, zmodel, kmodel, input_data, rtol=1e-5, atol=1e-5):
"""
Compare forward results for Keras model against Zoo Keras API model.
"""
WeightLoader.load_weights_from_kmodel(zmodel, kmodel)
zmodel.training(is_training=False)
bigdl_output = zmodel.forward(input_data)
keras_output = kmodel.predict(input_data)
self.assert_allclose(bigdl_output, keras_output, rtol=rtol, atol=atol)
def compare_model(self, zmodel, kmodel, input_data, rtol=1e-5, atol=1e-5):
"""
Compare forward results for Keras model against Zoo Keras API model.
"""
WeightLoader.load_weights_from_kmodel(zmodel, kmodel)
zmodel.training(is_training=False)
bigdl_output = zmodel.forward(input_data)
keras_output = kmodel.predict(input_data)
self.assert_allclose(bigdl_output, keras_output, rtol=rtol, atol=atol)
def compare_model(self, bmodel, kmodel, input_data, rtol=1e-5, atol=1e-5):
WeightLoader.load_weights_from_kmodel(bmodel, kmodel)
bmodel.training(is_training=False)
bigdl_output = bmodel.forward(input_data)
keras_output = kmodel.predict(input_data)
self.assert_allclose(bigdl_output, keras_output, rtol=rtol, atol=atol)
