task2_bilstm_config.json

{
  "data_dir": "data",
  "output_dir": "outputs",
  "embed_size": 256,
  "hidden_size": 512,
  "num_layers": 2,
  "dropout": 0.2,
  "batch_size": 2560,
  "lr": 0.001,
  "weight_decay": 1e-05,
  "num_epochs": 100,
  "seq_len": 32,
  "mask_prob": 0.15,
  "min_word_freq": 1,
  "early_stop_patience": 10,
  "grad_clip": 5.0,
  "checkpoint_path": "outputs/checkpoints/task2_bilstm_best.pt",
  "use_wandb": false,
  "wandb_project": "nlp-assignment3",
  "wandb_name": "task2_bilstm",
  "hf_repo_id": "nikhilrayaprolu/task2_bilstm",
  "hf_filename": "task2_bilstm.pt"
}