Enable test for stratified sampling for multi-class after load-balancing.
Merge branch 'master' into arno-deeplearning2