task1_lstm_config.json

{
  "data_dir": "data",
  "output_dir": "outputs",
  "embed_size": 64,
  "hidden_size": 256,
  "num_layers": 2,
  "dropout": 0.3,
  "batch_size": 1024,
  "lr": 0.001,
  "weight_decay": 1e-05,
  "num_epochs": 80,
  "early_stop_patience": 12,
  "grad_clip": 5.0,
  "checkpoint_path": "outputs/checkpoints/task1_lstm_best.pt",
  "use_wandb": true,
  "wandb_project": "nlp-assignment3",
  "wandb_name": "task1_lstm",
  "hf_repo_id": "nikhilrayaprolu/task1_lstm",
  "hf_filename": "task1_lstm.pt"
}