tbcnn-pytorch

TBCNN pytorch version, original tensorflow version on https://github.com/crestonbunch/tbcnn.

Configuration

set batch size, max tree size and max children size in proper value, not to exceed the gpu memory limitation.

python train.py

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
README.md		README.md
code_tokenizer.py		code_tokenizer.py
cuda_utils.py		cuda_utils.py
data.zip		data.zip
data_formatter.py		data_formatter.py
dataset.py		dataset.py
model.py		model.py
sublayer.py		sublayer.py
train.py		train.py
vocab_dict.py		vocab_dict.py
word_tokenizer.py		word_tokenizer.py