Pytorch转Mindspore,关于uniform_()方法求助
收藏回复举报
Pytorch转Mindspore,关于uniform_()方法求助
t('forum.solved') 已解决
发表于2024-01-26 12:53:30
0 查看

我在将Pytorch代码转成Mindspore代码,其中pytorch在nn.Module中有一部分代码是:

def init_parameters(self): 

        init_range = 0.1 

        for param in self.parameters(): 

            param.data.uniform_(-init_range, init_range)

我按照官方文档,将其中的self.parameters()改成了mindspore中nn.Cell模块里面的get_parameters()方法,返回的也是迭代器

现在的报错是 :AttributeError: 'Parameter' object has no attribute 'uniform_'

但我在文档里好像没找到uniform_()的替换,请教一下这里该怎么写呀?

我要发帖子