mindspore相关报错
mindspore重写WithLossCell,TrainOneStepCell接口错误:RuntimeError: _kernel.cc:88 CheckParam] AddN output shape must be equal to input shape.Trace: In file add_impl.py(272)/ return F.addn((x, y))/解决方案:不要在WithLossCell的construct方法中返回多个参数,否则就会在TrainOneStepCell的co
原创
2021-08-15 22:39:24 ·
269 阅读 ·
0 评论