megengine.amp.GradScaler.state_dict

GradScaler.state_dict()[源代码]