This will simplify later doing parameter sweeps over different models / training configurations. https://hydra.cc/docs/1.3/patterns/extending_configs/
