#

MNISTとのWGAN実験

9from labml import experiment
10
11from labml.configs import calculate

#

DCGAN 実験から構成をインポート

13from labml_nn.gan.dcgan import Configs

#

ヴァッサースタインGANの輸入損失

16from labml_nn.gan.wasserstein import GeneratorLoss, DiscriminatorLoss

#

ワッサースタインGAN損失の構成オプションを設定

19calculate(Configs.generator_loss, 'wasserstein', lambda c: GeneratorLoss())
20calculate(Configs.discriminator_loss, 'wasserstein', lambda c: DiscriminatorLoss())

#

23def main():

#

コンフィグオブジェクトの作成

25    conf = Configs()

#

実験を作成

27    experiment.create(name='mnist_wassertein_dcgan', comment='test')

#

オーバーライド設定

29    experiment.configs(conf,
30                       {
31                           'discriminator': 'cnn',
32                           'generator': 'cnn',
33                           'label_smoothing': 0.01,
34                           'generator_loss': 'wasserstein',
35                           'discriminator_loss': 'wasserstein',
36                       })

#

実験を開始してトレーニングループを実行する

39    with experiment.start():
40        conf.run()
41
42
43if __name__ == '__main__':
44    main()