我在将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_()的替换,请教一下这里该怎么写呀?
我在将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_()的替换,请教一下这里该怎么写呀?