NPSE on GPUs?

[enricobarausse]

hi all

I cannot seem to be able to run NPSE on GPUs, as some internal variables seem to be defined by default on CPUs:

return self.mean_t_fn(times) * x0
       ~~~~~~~~~~~~~~~~~~~~~~^~~~

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

ideas?

[janfb]

Hi @enricobarausse , this should now be solved and shipped with sbi v0.25.0. Please report if the issue has not been solved for your case.