Giters

KalmanNet / KalmanNet_TSP

code for KalmanNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

device problem

18181212864kq opened this issue · comments

commented

image
I encountered some difficulties when running main_linear.py. I tried changing some of the parameters (self.Q, self.R, and y) in Linear_KF.py to cpu, and the code ran successfully. However, I'm not sure if this is the correct approach.
It is the result.
image
Another issue is that when I train using CUDA on my 3060, it takes longer than training on my 5600G CPU. Why is this?
image

commented

Q1 Please use the updated main branch, where the device problem is solved.

Q2 When batch size increases, the advantage of GPU will become obvious.