addressed Madison and Ian's PR comments, except for testing this PR's impact on the performance of runing cifar10.yaml (it crashed on the master branch)