more support for multiple training sets