Hi!
I was trying to run the demo.ipynb but found that it does not work for me
RuntimeError Traceback (most recent call last)
<ipython-input-12-87b1f64074ed> in <module>
13 logit = model(X_train)
14 loss = criterion(logit.view(-1), y_train)
---> 15 loss += dp_loss(X_train, logit, sensi_train)
16 loss.backward()
17 optimizer.step()
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
Setting the selft.device to cpu fixes it but using gpu would be nice.
Great package btw :)
My system info:
-----
fairtorch NA
numpy 1.18.5
pydev_jupyter_vars NA
sinfo 0.3.1
torch 1.7.1
-----
IPython 7.19.0
jupyter_client 6.1.7
jupyter_core 4.7.0
notebook 6.1.5
-----
Python 3.8.5 (default, Jul 28 2020, 12:59:40) [GCC 9.3.0]
Linux-5.4.0-52-generic-x86_64-with-glibc2.29
12 logical CPU cores, x86_64
-----
Session information updated at 2021-01-12 13:20
Hi! I was trying to run the
demo.ipynb
but found that it does not work for meSetting the
selft.device
tocpu
fixes it but using gpu would be nice.Great package btw :)