config.json

{
    "dataset_name": "mschonhardt/georges-1913-normalization",
    "test_size": 0.2,
    "val_split": 0.5,
    "random_seed": 42,
    "special_tokens": {
      "<pad>": 0,
      "<sos>": 1,
      "<eos>": 2,
      "<unk>": 3
    },
    "model_parameters": {
      "embedding_dim": 64,
      "hidden_dim": 128,
      "num_layers": 3,
      "dropout": 0.3,
      "max_length": 100
    },
    "training_parameters": {
      "batch_size": 4096,
      "learning_rate": 0.0005,
      "epochs": 10,
      "teacher_forcing_ratio": 0.5,
      "clip_grad_norm": 1.0
    },
    "scheduler_parameters": {
      "scheduler_patience": 2,
      "scheduler_mode": "min",
      "scheduler_verbose": true
    },
    "file_paths": {
      "model_save_path": "normalization_model.pth",
      "vocab_save_path": "vocab.pkl"
    }
  }