9851098114134395edc8eb3a858a49bb25f95587,examples/chargan.py,,,#,88

Before Change

if args.action == "train":
    metrics = train(config, args)
    print("Resulting metrics:", metrics)
elif args.action == "search":
    search(config, args)
else:
    print("Unknown action: "+args.action)

After Change

inputs = TextInput(config, args.batch_size, one_hot=args.one_hot)

if args.action == "search":
    random_config = AlphaGANRandomSearch({}).random_config()

    if args.config_list is not None:
        config = random_config_from_list(args.config_list)

        config["generator"]=random_config["generator"]
        config["discriminator"]=random_config["discriminator"]
        // TODO Other search terms?
    else:
        config = random_config


def setup_gan(config, inputs, args):
    gan = hg.GAN(config, inputs=inputs)

    gan.create()

Instances