Register a customized BigDL module saver.
Register a customized BigDL module saver.
class name of the BigDL module
customized saver
Save a graph model to protobuf files so that it can be used in tensorflow inference.
Save a graph model to protobuf files so that it can be used in tensorflow inference.
When save the model, placeholders will be added to the tf model as input nodes. So you need to pass in the names and shape for the placeholders. BigDL model doesn't have such information. The order of the placeholder information should be same as the inputs of the graph model
graph model instance
placeholder information
where to save
model byte order
model data format
Save a graph model to protobuf files so that it can be used in tensorflow inference.
Save a graph model to protobuf files so that it can be used in tensorflow inference.
When save the model, placeholders will be added to the tf model as input nodes. So you need to pass in the names and shape for the placeholders. BigDL model doesn't have such information. The order of the placeholder information should be same as the inputs of the graph model
graph model instance
input node defs
where to save
model byte order