updated training to allow for easier mixing and matching of datasets and models