Changed training script to use new experiment config interface