分类：multiloss

2020
10-08

终于搞懂了Keras中multiloss的对应关系介绍

我就废话不多说了，大家还是直接看代码吧~model=Model(inputs=[src,tgt],outputs=[y,flow])#定义网络的时候会给出输入和输出model.compile(optimizer=Adam(lr=lr),loss=[losses.cc3D(),losses.gradientLoss('l2')],loss_weights=[1.0,reg_param])#训练网络的时候指定loss，如果是多loss，lossweights分别对应前面的每个loss的权重，最后输出loss的和train_loss=model.train_on_batch([X,atlas_vo... 继续阅读 >