问题描述:
请问pytorch的add_module函数对于的mindspore是如何表示的?
解答:
您好!可以用nn.CellList的append()函数替代
https://mindspore.cn/docs/api/zh-CN/r1.6/api_python/nn/mindspore.nn.CellList.html?highlight=celllist
example:
conv = nn.Conv2d(100, 20, 3)
bn = nn.BatchNorm2d(20)
relu = nn.ReLU()
cell_ls = nn.CellList([bn])
cell_ls.insert(0, conv)
cell_ls.append(relu)
另外你看源码mindspore\python\mindspore\nn\cell.py的893行,可以发现insert_child_to_cell()其实就是给字典类型的self._cells增加了一个键值对