diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index bd0b926a..ce315685 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -351,7 +351,6 @@ def to_onnx(graph: backend.GraphHandler, name: str) -> ModelProto: def push_output(self, name: str, tensor: backend.Tensor) -> str: self.names[tensor] = name - # TODO 需要判断全图输出并保存到 outputs return name def push_input(self, tensor: backend.Tensor) -> str: diff --git a/src/ffi/ffi_infinitensor.cc b/src/ffi/ffi_infinitensor.cc index 09fa361c..e545f43c 100644 --- a/src/ffi/ffi_infinitensor.cc +++ b/src/ffi/ffi_infinitensor.cc @@ -29,7 +29,7 @@ void export_values(py::module &m) { #define VALUE(TYPE, NAME) value(#NAME, TYPE::NAME) py::enum_(m, "ActType") - .value("Linear", ActType::None) // None 是 Python 关键字,不能用 + .value("Linear", ActType::None) // `None` is Python keyword .VALUE(ActType, Relu) .VALUE(ActType, Sigmoid) .VALUE(ActType, Tanh)