task2_ssm_config.json

{
  "data_dir": "data",
  "output_dir": "outputs",
  "embed_size": 256,
  "d_model": 512,
  "state_size": 64,
  "num_layers": 4,
  "dropout": 0.1,
  "batch_size": 640,
  "lr": 0.001,
  "weight_decay": 1e-05,
  "num_epochs": 60,
  "seq_len": 32,
  "min_word_freq": 1,
  "early_stop_patience": 10,
  "grad_clip": 5.0,
  "checkpoint_path": "outputs/checkpoints/task2_ssm_best.pt",
  "use_wandb": true,
  "wandb_project": "nlp-assignment3",
  "wandb_name": "task2_ssm",
  "hf_repo_id": "nikhilrayaprolu/task2_ssm",
  "hf_filename": "task2_ssm.pt"
}