DataParallel on GPUs not supported? about pytorch_wavelets HOT 5 CLOSED

I also get the same error when using DataParallel, good find. This is something I haven't checked for before. Will try fix it now.

from pytorch_wavelets.

Actually, the following modification works:
`self.h = [t.to(ll.device) for t in self.h]

y = lowlevel.afb2d(ll, self.h, self.mode)`

from pytorch_wavelets.

I didn't see your code, but I've done something very similar. Thanks!

from pytorch_wavelets.

Actually, the following modification works: `self.h = [t.to(ll.device) for t in self.h]

y = lowlevel.afb2d(ll, self.h, self.mode)`

I don't understand the meaning of these two lines of code. Can you give a more detailed explanation? Thank u very much!

from pytorch_wavelets.

I think this code was just casting all the bandpass to the device. Did you have an issue with DataParallel?

from pytorch_wavelets.