From 62fd6199874f67dec09d690df0a757084151647e Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Wed, 15 Mar 2023 15:16:16 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E7=A7=BB=E9=99=A4=E4=B8=AD=E6=96=87?= =?UTF-8?q?=E6=B3=A8=E9=87=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YdrMaster --- pyinfinitensor/src/pyinfinitensor/onnx.py | 1 - src/ffi/ffi_infinitensor.cc | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) 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)