在pytorch迁移至mindspore过程中,发现在迁移后的mindspore的SequentialCell重复模块中,相同的网络结构给定相同的维度输入,在construct中输出了前后维度对比,发现输出维度不一致。
【操作步骤&问题现象】
相同维度的输入至相同的网络,输出维度不一致,导致tensor的加法报错
在construct中输出了经过SequentialCell前后的维度,发现维度不一致问题
两两为一组,第一个为输入construct的维度,第二个为经过网络模型的输出维度,发现到第三个模块的时候维度突然变的不一致
网络结构为这样相同的模块重复8次,在第3个重复模块中出现问题,跟原pytorch模型对应过网络权重,能够对的上,求助是网络结构迁移存在问题还是其他问题。
【截图信息】
pytorch模型结构
迁移mindspore结构
可以打开看一下ResnetBlock_remove_IN这个结构里面shape的变化.