pytorch迁移到mindspore,pytorch中的nn.ModuleDict()类应该如何进行映射
收藏回复举报
pytorch迁移到mindspore,pytorch中的nn.ModuleDict()类应该如何进行映射
t('forum.solved') 已解决
新人帖
发表于2023-09-28 16:50:59
0 查看
在迁移框架的时候,pytorch中使用了nn.ModuleDict()这个类,该类可以把所有子模块放到一个有序字典中,里面还定义了诸如keys,values,update的方法。 

但在MindSpore 中没有找到可以直接映射或者相类似的类可以使用,如果重写这个类的话感觉难度有点大,因为ModuleDict()就是一个底层的容器类。请问有没有好的解决办法

我要发帖子