search

jonkhler / s2cnn

Spherical CNNs
MIT License
939 stars 176 forks source link

maximum bandwidth support #31

Closed wuchlei closed 5 years ago

wuchlei commented 5 years ago

It seems that the current SO3Convolution and S2Convolution only support small bandwidth, e.g., b_in = 64. Because when I change the bandwidth to 128 or 256, it's either memory error or overflow error. Am I doing something wrong?

Please help.

wuchlei commented 5 years ago

For anyone else that has a lot of out-of-memory errors, change the bandwidth of so3 convolution to a very small value, e.g., 10, as they did in the examples. I think the current implementation only supports the low bandwidth convolutions.