Adding UI to provide config variables during training set process.