pytorch-tree-lstm
pytorch-tree-lstm copied to clipboard
Questions about batching
Great work!
My question is: with batch_tree_input and train_data_generator, it's possible to train a batch of trees right? Could you please provide an example how to generate the TreeDataset() in DataLoader and how to use train_data_generator?
Thank you!
I also want to know how to construct the tree using normal vector data