TBCNN pytorch version, original tensorflow version on https://github.com/crestonbunch/tbcnn.
set batch size, max tree size and max children size in proper value, not to exceed the gpu memory limitation.
python train.py
TBCNN pytorch version, original tensorflow version on https://github.com/crestonbunch/tbcnn.
set batch size, max tree size and max children size in proper value, not to exceed the gpu memory limitation.
python train.py