pytorch_spline_conv icon indicating copy to clipboard operation
pytorch_spline_conv copied to clipboard

Published 20 hours ago • verified publisher icon rusty1s

Batch operation support

Open wdr123 opened this issue 1 year ago • 2 comments

Hi authors, thanks for this amazing work. My question is when I try to put the model's parameter in cuda mode, the spline_conv doesn't work and inform me parameters (weights) must be in cpu mode. Also spline_conv seems doesn't support batch as the input tensor's first dimension like other NN's forwarding function. Is that true? If it is, how can we achieve data parallel on GPU with spline_conv(.)? Thanks.

wdr123 avatar Apr 25 '24 20:04 wdr123