config dictionary for baseline.yml has no batch_norm

[NinaM31]

when running this in the terminal:

python train.py --model_kind baseline --config config/baseline.yml

I get this:

Traceback (most recent call last):
  File "C:\Users\ninam\Desktop\Arabic_Diacritization\train.py", line 43, in <module>
    trainer = CBHGTrainer(args.config, args.model_kind)
  File "C:\Users\ninam\Desktop\Arabic_Diacritization\trainer.py", line 51, in __init__
    self.model = self.config_manager.get_model()
  File "C:\Users\ninam\Desktop\Arabic_Diacritization\config_manager.py", line 217, in get_model
    return self.get_baseline()
  File "C:\Users\ninam\Desktop\Arabic_Diacritization\config_manager.py", line 225, in get_baseline
    use_batch_norm=self.config["use_batch_norm"],
  File "C:\Users\ninam\miniconda3\envs\pyTorch3.9\lib\site-packages\ruamel\yaml\comments.py", line 927, in __getitem__
    return ordereddict.__getitem__(self, key)
KeyError: 'use_batch_norm'