这个问题之前问过一遍了,不过还没有解决。
mindspore-alpha2.0, ubuntu,gpu,pytorch迁移到mindspore
之前迁移过一个单个网络的模型,对于单个网络训练我知道如何实现,如下:

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

原pytorch代码如下:




就是图像数据通过feature_encoder得到feature,feature在分别输入domain_classifier和metric_net_encoder来得到loss,通过loss.forward不断训练更新feature_encoder,最终目的是要训练feature_encoder。
但是这个三个网络协同训练的我就不知道咋迁移了
这个问题之前问过一遍了,不过还没有解决。
mindspore-alpha2.0, ubuntu,gpu,pytorch迁移到mindspore
之前迁移过一个单个网络的模型,对于单个网络训练我知道如何实现,如下:
但是目前迁移的这个有三个网络模块协同训练,结构图如图:
原pytorch代码如下:
就是图像数据通过feature_encoder得到feature,feature在分别输入domain_classifier和metric_net_encoder来得到loss,通过loss.forward不断训练更新feature_encoder,最终目的是要训练feature_encoder。
但是这个三个网络协同训练的我就不知道咋迁移了