| { | |
| "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" | |
| } |