The original implementation for paper Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks doesn't support batch calculation of TreeLSTM.
To run the model with batch TreeLSTM:
- python main.py --use_batch --batchsize 25
which should give you exact results as without batch, but much faster in training and inference.