hongshaorou/YOPO
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix a warning issue and update readme

Browse Source
This commit is contained in:
Lu Junjie
2024-11-02 12:25:00 +08:00
parent 079d8cf23a
commit 8142087649
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flightpolicy/envs/vec_env_wrapper.py
View File

@@ -34,7 +34,7 @@ class FlightEnvVec(VecEnv):
self._gray_img_obs = np.zeros([self.num_envs, self.img_width * self.img_height], dtype=np.uint8)
self._depth_img_obs = np.zeros([self.num_envs, self.img_width * self.img_height], dtype=np.float32)
self._reward = np.zeros([self.num_envs, self.reward_dim], dtype=np.float32)
self._done = np.zeros((self.num_envs), dtype=np.bool)
self._done = np.zeros((self.num_envs), dtype=bool)
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# observation: [p_wb, v_b, a_b, q_wb] (in Body Frame); action: dp_pred; reward: cost