pytorch中的register_buffer方法如何映射到mindspore中
收藏回复举报
pytorch中的register_buffer方法如何映射到mindspore中
t('forum.solved') 已解决
发表于2023-11-19 17:17:56
0 查看

在进行pytorch框架迁移到mindspore时,pytorch的代码使用了register_buffer()方法,该方法定义在pytorch中的Module类中,其作用为向模型中添加一个buffer,通常用于为不应被视为模型参数的信息记录一个buffer。在mindspore的2.2版本中,没有找到有关该方法的映射说明,在官方文档中也没有找到相似方法进行替代,请问有无解决办法?

我要发帖子