问题描述:
【功能模块】
mindspore版本1.6
【操作步骤&问题现象】
1、我在做自定义算子时遇到了一个问题,pytorch中有module.apply函数一次性对所有子模块初始化权重赋值,但msp中没有cell.apply
2、所以msp中怎么实现相应方法呢
【截图信息】
pytorch代码:
minspore代码
解答:
具体的实现方法:https://www.cnblogs.com/sharklet/p/14983398.html
问题描述:
【功能模块】
mindspore版本1.6
【操作步骤&问题现象】
1、我在做自定义算子时遇到了一个问题,pytorch中有module.apply函数一次性对所有子模块初始化权重赋值,但msp中没有cell.apply
2、所以msp中怎么实现相应方法呢
【截图信息】
pytorch代码:
minspore代码
解答:
具体的实现方法:https://www.cnblogs.com/sharklet/p/14983398.html