Open dongliangcao opened 2 years ago
Hi Dongliang,
Glad to hear you're interested!
There isn't currently any code built-in to this repo for training on irregular-sized batches.
There are two main strategies we could follow here:
If we were going to add something to this repo, I'd probably lean towards option (A). We could add some utilities to make it easy to pad out irregular inputs, and double-check that all operations behave sanely with respect to padding.
Let me know if that sounds reasonable! I probably wouldn't get to implementing anything until Feb at the earliest, I'm pretty busy at the moment.
Dear author,
Thanks to your great work and code release for DiffusionNet. When I am working with DiffusionNet, I am wondering if there is a solution to train DiffusionNet with batch size larger than 1, even though the shape data might have different number of vertices.
Best regards, Dongliang