华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
在进行pytorch框架迁移到mindspore时,pytorch的代码使用了register_buffer()方法,该方法定义在pytorch中的Module类中,其作用为向模型中添加一个buffer,通常用于为不应被视为模型参数的信息记录一个buffer。在mindspore的2.2版本中,没有找到有关该方法的映射说明,在官方文档中也没有找到相似方法进行替代,请问有无解决办法?
我要发帖子
在进行pytorch框架迁移到mindspore时,pytorch的代码使用了register_buffer()方法,该方法定义在pytorch中的Module类中,其作用为向模型中添加一个buffer,通常用于为不应被视为模型参数的信息记录一个buffer。在mindspore的2.2版本中,没有找到有关该方法的映射说明,在官方文档中也没有找到相似方法进行替代,请问有无解决办法?