54a5a577d06acce4d13aa8d564c85e8d57ac00e7,research/gan/progressive_gan/train_main.py,,,#,37

Before Change

flags.DEFINE_bool("to_rgb_use_tanh_activation", False,
                  "Whether to apply tanh activation when output rgb.")

flags.DEFINE_integer("batch_size", 8, "Number of images in each batch.")

flags.DEFINE_integer("stable_stage_num_images", 1000,
                     "Number of images in the stable stage.")

After Change

flags.DEFINE_integer("num_resolutions", 4, "Number of progressive resolutions.")

flags.DEFINE_list(
    "batch_size_schedule", [8, 8, 4],
    "A list of batch sizes for each resolution, if "
    "len(batch_size_schedule) < num_resolutions, pad the schedule in the "
    "beginning with the first batch size.")

flags.DEFINE_integer("kernel_size", 3, "Convolution kernel size.")

flags.DEFINE_integer("colors", 3, "Number of image channels.")

Instances