We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 51c4579 commit 601f79cCopy full SHA for 601f79c
1 file changed
mltu/tensorflow/callbacks.py
@@ -24,7 +24,7 @@ def __init__(
24
def on_train_end(self, logs=None):
25
self.model.load_weights(self.saved_model_path)
26
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)
+ tf2onnx.convert.from_keras(self.model, output_path=self.onnx_model_path)
28
29
if self.metadata and isinstance(self.metadata, dict):
30
# Load the ONNX model
0 commit comments