You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Traceback (most recent call last):
File "/home/wfang/spikingjelly_dev/spikingjelly/test4.py", line 15, in<module>
net(x)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/lava/lib/dl/slayer/neuron/cuba.py", line 439, in forward
_, voltage = self.dynamics(input)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/lava/lib/dl/slayer/neuron/cuba.py", line 365, in dynamics
current = self.norm(current)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/lava/lib/dl/slayer/neuron/norm.py", line 209, in forward
std = self.std(var)
File "/home/wfang/anaconda3/envs/lava-env/lib/python3.10/site-packages/lava/lib/dl/slayer/neuron/norm.py", line 170, in std
return torch.ones(1) <<torch.ceil(torch.log2(std)).clamp(RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
(lava-env) wfang@mlg-ThinkStation-P920:~$ conda list lava
# packages in environment at /home/wfang/anaconda3/envs/lava-env:## Name Version Build Channel
lava 0.5.0 pyhd8ed1ab_0 conda-forge
lava-dl 0.3.0 pyhd8ed1ab_0 conda-forge
(lava-env) wfang@mlg-ThinkStation-P920:~$ conda list torch
# packages in environment at /home/wfang/anaconda3/envs/lava-env:## Name Version Build Channel
pytorch 1.12.1 cuda112py310h51fe464_200 conda-forge
torchvision 0.13.0 cuda112py310h453157a_0 conda-forge
The text was updated successfully, but these errors were encountered:
Objective of issue:
Lava DL version:
Lava version:
I'm submitting a ...
Current behavior:
Expected behavior:
Steps to reproduce:
Run the following codes:
Related code:
Other information:
My env:
The text was updated successfully, but these errors were encountered: