新手入门
用户指南
API 参考
开发者指南
论坛
官网

GitHub
Bilibili

megengine
megengine.data
megengine.functional
megengine.module
megengine.module.init
megengine.autodiff
megengine.optimizer
megengine.jit
megengine.amp
megengine.dtr
megengine.distributed
megengine.quantization
megengine.random
megengine.traced_module
megengine.hub
megengine.utils

API 相关说明

与其它框架 API 进行对比
Deprecated APIs
环境变量（Env）设置

megengine.amp.GradScaler.load_state_dict¶

GradScaler.load_state_dict(state)[源代码]¶

上一页

megengine.amp.GradScaler.backward

下一页

megengine.amp.GradScaler.state_dict