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)