megengine.jit

>>> from megengine.jit import trace

trace

包装一个可调用对象，并提供以下功能：

Trace Config

SublinearMemoryConfig	亚线性内存优化的配置。
DTRConfig	Configuration for DTR memory optimization.
GraphOptimizationConfig	Configuration for graph optimization: False for OFF, True for ON.

上一页

megengine.optimizer.clip_grad_value

下一页

megengine.jit.trace