Browse Source

docs(imperative/jit): add docstring of dump

GitOrigin-RevId: 49eacc0e58
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
2afa0af926
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/jit/tracing.py

+ 2
- 1
imperative/python/megengine/jit/tracing.py View File

@@ -1060,7 +1060,8 @@ class trace:
resize_input: whether resize input image to fit input var shape.
input_transform: a python expression to transform the input data.
Example: data / np.std(data)
dump_format: using different dump formats.
dump_format: using different dump formats. the open source MegEngine defaults to the FBS
format. internal MegEngine have a choice of FBS and internal proprietary formats

Keyword Arguments:



Loading…
Cancel
Save