-
Notifications
You must be signed in to change notification settings - Fork 4
TreeLSTM guide
Egor Spirin edited this page Feb 16, 2021
·
1 revision
PyTorch reimplementation of ChildSum version of TreeLSTM presented by Tai et al. in "Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks".
A full list of train parameters is declared by the .yaml config. Examples of configs are store in the config folder.
To train the model with your config use
python train_treelstm.py -cn <config name>