MNISTසමඟ WGAN අත්හදා බැලීම

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

13from labml_nn.gan.dcgan import Configs

16from labml_nn.gan.wasserstein import GeneratorLoss, DiscriminatorLoss

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()