def __init__(self, model_conf, net_def, model_weights,
model_name, offset16=False, write_magic=False):
self.model_conf = model_conf
data_type = model_conf.get(ModelKeys.data_type, mace_pb2.DT_FLOAT)
# self.net_def.arg
if model_conf.get(ModelKeys.quantize_schema) == "int8":
data_type = mace_pb2.DT_INT8
self.net_def = MicroIoConverter.convert(net_def, data_type)
self.model_weights = model_weights
self.model_name = model_name
self.offset16 = offset16
self.write_magic = write_magic
self.code_gen = MicroCodeGen()
self.np_data_type = data_type_to_np_dt(data_type, np.float32)
self.model_dir = "micro/codegen/" + model_name + "/"
util.mkdir_p(self.model_dir)
self.op_resolver = OpResolver(self.net_def, self.model_conf)
def __init__(self,
model_conf,
net_def,
model_weights,
model_name,
offset16=False,
write_magic=False):
self.model_conf = model_conf
data_type = model_conf.get(ModelKeys.data_type, mace_pb2.DT_FLOAT)
self.net_def = MicroIoConverter.convert(net_def, data_type)
self.model_weights = model_weights
self.model_name = model_name
self.offset16 = offset16
self.write_magic = write_magic
self.code_gen = MicroCodeGen()
data_type = model_conf.get(ModelKeys.data_type, mace_pb2.DT_FLOAT)
self.np_data_type = data_type_to_np_dt(data_type, np.float32)
self.gen_folder = 'micro/codegen/'
util.mkdir_p(self.gen_folder)
self.op_resolver = OpResolver(self.net_def, self.model_conf)