config.json

{
  "images_dir": "data/raw/Images",
  "processed_dir": "data/processed",

  "encoder_dim": 2048,
  "embed_size": 300,
  "hidden_size": 1024,
  "attention_dim": 256,
  "dropout": 0.5,
  "rnn_type": "gru",

  "alpha_c": 1.0,

  "batch_size": 256,
  "num_workers": 16,
  "num_epochs": 35,
  "decoder_lr": 3.2e-3,
  "encoder_lr": 8e-5,
  "weight_decay": 0.0,
  "grad_clip": 5.0,
  "scheduler_patience": 3,
  "scheduler_factor": 0.5,

  "fine_tune_start_epoch": 10,
  "fine_tune_blocks": 2,

  "label_smoothing": 0.1,
  "scheduled_sampling_max": 0.25,

  "seed": 42,
  "save_dir": "models",
  "run_name": "attention_gru_glove",
  "log_interval": 50,
  "val_bleu_subset": 500,
  "wandb_project": "flickr-captioning",
  "wandb_mode": "online",

  "glove_path": "data/glove/glove.6B.300d.txt"
}