More complete version of the code

Hello,
I was wondering if you are planning to complete the current code, or I am missing something. 
The main function has TreeLSTM as the model which is not defined. I also wanted to know how should the trainloader be.