4

我正在研究 GANS，我需要在下班后保存模型。然后我必须再次重新训练之前保存的模型。我正在保存这三个模型以供以后继续训练。

Discriminator Model.h5
Generator Model.h5
Generator-on-Discriminator Model.h5

对于这些模型，我使用perceptual lossand Wasserstein loss。但是当我load_model再次重新训练保存的模型时，它会遇到以下错误。

Unknown loss function:wasserstein_loss

我也尝试过Discriminator.compile(loss=Wasserstein loss)，但这仍然不能解决我的问题。你们中的任何人都可以指导我解决这个问题，并告诉我可以使用 train_on_batch() 重新训练保存的模型。

4

1 回答 1

3

我自己解决了

在加载模型时定义custom_objects={'wassertein_loss':wassertein_loss}路径解决了我的问题。IE

Discriminator=load_model(model_path, custom_objects={'wassertein_loss':wassertein_loss} )
于 2019-01-10T09:27:54.863 回答