dnn trainer fixes