crestonbunch / tbcnn

Efficient tree-based convolutional neural networks in TensorFlow
MIT License
150 stars 44 forks source link

pytorch implementation #17

Open gkoumasd opened 3 years ago

gkoumasd commented 3 years ago

Hi,

I am looking for a PyTorch version of tbcnn. I cannot find a PyTorch implementation on google. I wonder if you have completed your PyTorch implementation and if you could share it with me.

Tomsawyerhu commented 1 year ago

hey, maybe you can check https://github.com/Tomsawyerhu/tbcnn-pytorch, it's a pytorch version for classification tasks.