python代码的日志记录

import logging

logging.basicConfig(level=logging.INFO)
logging.getLogger("ModelHelper").setLevel(logging.INFO)
log = logging.getLogger("ModelHelper")

log.info("Number of FPN output channels is {}".format(self.fpn_out_channels))

try:
    self.graph.fold_constants(fold_shapes=True)
except TypeError as e:
    log.error("This version of ONNX GraphSurgeon does not support folding shapes, please upgrade your " "onnx_graphsurgeon module. Error:\n{}".format(e))
    raise