多个网络协同训练方法报错
收藏回复举报
多个网络协同训练方法报错
t('forum.solved') 已解决
发表于2023-05-19 11:40:19
0 查看

这个问题之前问过一遍了,不过还没有解决。

mindspore-alpha2.0, ubuntu,gpu,pytorch迁移到mindspore

之前迁移过一个单个网络的模型,对于单个网络训练我知道如何实现,如下:

cke_2175.png

但是目前迁移的这个有三个网络模块协同训练,结构图如图:

cke_7433.png

原pytorch代码如下:

cke_10916.png

cke_11686.png

cke_12694.png

cke_13807.png

就是图像数据通过feature_encoder得到feature,feature在分别输入domain_classifier和metric_net_encoder来得到loss,通过loss.forward不断训练更新feature_encoder,最终目的是要训练feature_encoder。

但是这个三个网络协同训练的我就不知道咋迁移了

我要发帖子