4

私は GANS に取り組んでおり、勤務時間後にモデルを保存する必要があります。そして、以前に保存したモデルを、保存された場所で再トレーニングする必要があります。後でトレーニングを続けるために、これら 3 つのモデルを保存しています。

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

これらのモデルには、 と を使用perceptual lossしてWasserstein lossいます。しかし、load_model保存したモデルを再度トレーニングすると、次のエラーが発生します。

Unknown loss function:wasserstein_loss

私も試しDiscriminator.compile(loss=Wasserstein loss)ましたが、これでも問題は解決しません。誰かがこれについて私を案内してくれませんか、train_on_batch() を使用して保存されたモデルを再トレーニングすることが可能であることを教えてください。

4

1 に答える 1