Skip to content

Commit 601f79c

Browse files
committed
should be tf2onnx instead of self.tf2onnx
1 parent 51c4579 commit 601f79c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mltu/tensorflow/callbacks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ def __init__(
2424
def on_train_end(self, logs=None):
2525
self.model.load_weights(self.saved_model_path)
2626
self.onnx_model_path = self.saved_model_path.replace(".h5", ".onnx")
27-
self.tf2onnx.convert.from_keras(self.model, output_path=self.onnx_model_path)
27+
tf2onnx.convert.from_keras(self.model, output_path=self.onnx_model_path)
2828

2929
if self.metadata and isinstance(self.metadata, dict):
3030
# Load the ONNX model

0 commit comments

Comments
 (0)