search

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
MIT License
2.67k stars 377 forks source link

Problem on Running a3tgcn_example.py #259

Open netlabcode opened 1 year ago

netlabcode commented 1 year ago

I tried to run example codes, most of them are working. I am running using GPU. However, when I run a3tgcn_example.py, I got the following error:

_H_accum = H_accum + probs[period] * self._basetgcn( RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

What possible solution for this issue?

EVEREST-dlk commented 11 months ago

I also encountered this problem, and how did you solve it?

netlabcode commented 11 months ago

I also encountered this problem, and how did you solve it?

I am still unable to resolve the issue.