ParamBase 类型操作的相关问题
l430h682 发布于2021-12 浏览:1493 回复:1
0
收藏

这个是想根据层的名字来对参数进行初始化,请问应该如何解决呢

收藏
点赞
0
个赞
共1条回复 最后由l430h682回复于2021-12
#2l430h682回复于2021-12

修改为

m.__class__.weight_attr = nn.initializer.Normal(0.0, 0.02)
0
TOP
切换版块