There is an error, when run "python train_ppo.py"

jzstudent / UAV-auto-navigation-and-object-tracking-based-on-RL

毕业设计的代码部分，实现了UE4和airsim环境下无人机自主导航和目标跟踪的强化学习算法。

201 stars 28 forks source link

There is an error, when run "python train_ppo.py" #10

Open xiaoToby opened 6 months ago

xiaoToby commented 6 months ago

The error log: File "d:/UAV-auto-navigation-and-object-tracking-based-on-RL-main/airsim_rl/trainppo.py", line 120, in main rollout.obs[0].copy(torch.tensor(ob)) RuntimeError: The size of tensor a (112) must match the size of tensor b (256) at non-singleton dimension 3

jzstudent commented 5 months ago

check the config json file of airsim, this problem is casued by the image size