def load_model_bigdl(self):
fp = open("bigdl_model.model", 'wb')
fp.write(self.model)
fp.close()
from zoo.pipeline.inference import InferenceModel
model = InferenceModel()
model.load_bigdl("bigdl_model.model")
return model
def test_load_bigdl(self):
model = InferenceModel(3)
model.load_bigdl(
os.path.join(resource_path, "models/bigdl/bigdl_lenet.model"))
input_data = np.random.random([4, 28, 28, 1])
output_data = model.predict(input_data)