search

ranahanocka / MeshCNN

Convolutional Neural Network for 3D meshes in PyTorch
MIT License
1.58k stars 316 forks source link

torch.BoolTensor is supported since torch >= 1.2.0 #25

Closed ycjungSubhuman closed 5 years ago

ycjungSubhuman commented 5 years ago

Hi, first of all, thanks for sharing this code public!

I've discovered a problem with the README. It says it works with pytorch 1.0, but it won't because it is using torch.BoolTensor in model/layers/mesh.py. torch.BoolTensor is supported since 1.2.0.

I confirmed the test code for scripts/human_seg is working fine with pytorch 1.2.0. I hope this information helps other guys testing this code.

ranahanocka commented 5 years ago

Hi @ycjungSubhuman ,

Thanks , I updated it :)