Skip to content

Commit

Permalink
set ort logger
Browse files Browse the repository at this point in the history
  • Loading branch information
SkyTNT committed Oct 8, 2024
1 parent 5131874 commit 41b5346
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions app_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@
VERSION = "v1.3.3"
MAX_SEED = np.iinfo(np.int32).max

rt.set_default_logger_severity(3)


def softmax(x, axis):
x_max = np.amax(x, axis=axis, keepdims=True)
Expand Down

0 comments on commit 41b5346

Please sign in to comment.