microsoft / O-CNN

O-CNN: Octree-based Convolutional Neural Networks for 3D Shape Analysis

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there any plan for implementing multi-gpu version of pytorch model?

SuekyeongNam opened this issue · comments

I realized that there is no module for multi-gpu because of the structure of octree.
Is there any plan for implementing multi-gpu version of pytorch model?

Additionally, do you have a plan for publishing autoencoder model using pytorch?

In theory, the multi-gpu mode can be implemented. And I will release the code in the near future.