def save_to_db(db_type, db_destination, predictor_export_meta): meta_net_def = get_meta_net_def(predictor_export_meta) workspace.FeedBlob(predictor_constants.META_NET_DEF, serde.serialize_protobuf_struct(meta_net_def)) blobs_to_save = [predictor_constants.META_NET_DEF] + \ predictor_export_meta.parameters op = core.CreateOperator("Save", blobs_to_save, [], absolute_path=True, db=db_destination, db_type=db_type) workspace.RunOperatorOnce(op)
def save_to_db(db_type, db_destination, predictor_export_meta): meta_net_def = get_meta_net_def(predictor_export_meta) with core.DeviceScope(core.DeviceOption(caffe2_pb2.CPU)): workspace.FeedBlob( predictor_constants.META_NET_DEF, serde.serialize_protobuf_struct(meta_net_def) ) blobs_to_save = [predictor_constants.META_NET_DEF] + \ predictor_export_meta.parameters op = core.CreateOperator( "Save", blobs_to_save, [], absolute_path=True, db=db_destination, db_type=db_type) workspace.RunOperatorOnce(op)
def save_to_db(db_type, db_destination, predictor_export_meta, use_ideep=False): meta_net_def = get_meta_net_def(predictor_export_meta) device_type = caffe2_pb2.IDEEP if use_ideep else caffe2_pb2.CPU with core.DeviceScope(core.DeviceOption(caffe2_pb2.CPU)): workspace.FeedBlob(predictor_constants.META_NET_DEF, serde.serialize_protobuf_struct(meta_net_def)) blobs_to_save = [predictor_constants.META_NET_DEF] + \ predictor_export_meta.parameters op = core.CreateOperator("Save", blobs_to_save, [], device_option=core.DeviceOption(device_type), absolute_path=True, db=db_destination, db_type=db_type) workspace.RunOperatorOnce(op)