about the input of he CNN_net

mbrossar / ai-imu-dr

AI-IMU Dead-Reckoning

MIT License

788 stars 222 forks source link

about the input of he CNN_net #50

Open Iqun1314 opened 3 years ago

Iqun1314 commented 3 years ago

Hello, I tried to replace the input with wx, wy, wz, ax, ay, az, roll, pitch, yall, the code is: if iekf.mes_net.training: u = dataset.add_noise(u) u = u.numpy()
u = np.hstack((u, ang_gt)) #6+3=9 u = torch.from_numpy(u) but when I run the code, there is an error: RuntimeError: Given groups=1, weight of size 32 9 5, expected input[1, 6, 2000] to have 9 channels, but got 6 channels instead

I don't know how to deal with it, could you give me some suggestion, thank you!

mbrossar commented 3 years ago

You give an input of dimension 9 but the network expects an input of dimension 6. It is sufficient to modify the input dimension (the 6) here.

scott81321 commented 1 year ago

@Iqun1314 Hello! Have you had success so far? Can anyone confirm if, whether or not, the training part of ai-imu-dr only considers updating the NN variable of the method?