如何在Megengine 1.5.0中高效实现ModelEMA功能?

为使您的问题得到快速解决,建议选择对应标签。
大家好,我想用Megengine 1.5.0实现ModelEMA的功能,遇到几个问题:

1. 如何正确获取模型参数的索引?
我试过model.state_dict(keep_var=True),model.named_parameters(),model.named_buffers(),哪一个是推荐的操作。

2.如何更新EMA模型的权重?
在Megengine 1.4.0中有一个_inplace_add_的函数,但是在1.5.0中使用会报错;
使用Tensor._reset()函数,感觉比较慢;
使用moving *= decay; moving += (1-d) * new;