• Gan Quan's avatar
    Fix 0deg update_to and Tree-LSTM model (#51) · b7eb1659
    Gan Quan authored
    * WIP
    
    * WIP
    
    * treelstm dataloader
    
    * Main training loop.
    
    * trainable treelstm script
    
    * fix dependency
    
    * cuda training
    
    * Add tensorized topological traversal
    
    * allowing update_to() with no incoming messages
    
    * fixing partial cases
    b7eb1659
test_basics.py 3.03 KB