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

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

Opened this issue · 1 comments

The error log:
File "d:/UAV-auto-navigation-and-object-tracking-based-on-RL-main/airsim_rl/train_ppo.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

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