search

KalmanNet / KalmanNet_TSP

code for KalmanNet
287 stars 76 forks source link

device problem #19

Open 18181212864kq opened 1 year ago

18181212864kq commented 1 year ago

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

KalmanNet commented 1 year ago

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.