remove --max-depth-length option in the train function in train.sh