config.json

{
  "train" : {
    "dataset" : "../../data/cnn_pickled/cnn_pointer_50k",
    "num_articles" : -1,
    "num_evaluate" : 13000,
    "throw" : 0,
    "n_epochs" : 1,
    "batch_size" : 50,
    "discriminator_batch_size" : 32,
    "generator_learning_rate" : 0.0001,
    "discriminator_learning_rate" : 0.0001,
    "beta" : 1.00,
    "lambda": 0.75,
    "n_generator" : 4,
    "n_discriminator" : 0,
    "discriminator_n_epochs" : 0,
    "discriminator_fake_data_sample_rate" : 1.00,
    "num_monte_carlo_samples" : 4,
    "max_sample_length" : 99,
    "sample_rate" : 0.80,
    "allow_negative_reward" : true,
    "use_trigram_check" : false,
    "use_running_avg_baseline": false
  },
  "evaluate" : {
    "expansions" : 3,
    "keep_beams" : 20,
    "return_beams": 5
  },
  "generator_model" : {
    "embedding_size" : 100,
    "n_layers" : 1,
    "hidden_size" : 128,
    "dropout_p" : 0.0,
    "load" : true,
    "load_file" : "../../models/pretrained_models/cnn/epoch13_cnn_test1.pth.tar"
  },
  "discriminator_model" : {
    "hidden_size" : 128,
    "dropout_p" : 0.5,
    "num_kernels" : 100,
    "kernel_sizes" : [3, 4, 5],
    "load" : true,
    "load_file" : "../../models/pretrained_models/classifier/cnn/epoch10_dataset_num100_cnn_classifier_sampled_folder_test5.tar"
  },
  "save" : {
    "save_file_generator" : "cnn_generator_seqGAN_start_sr_0.80.pth.tar",
    "save_file_discriminator" : "cnn_discriminator_seqGAN_start_sr_0.80.pth.tar"
  },
  "log" : {
    "print_every" : 200,
    "filename": "experiments/cnn_test_1/output_gantest_1.log"
  },
  "tensorboard" : {
    "log_path" : "../../log/GAN/seqGAN_start_sr_0.80"
  }
}