StopIteration Error

[XinyiChu]

When I run python train.py . It shows:
Traceback (most recent call last):
File "/workspace/vo_rl/env/svo_wrapper.py", line 85, in get_images_pose
batch = next(self.dataloader_iter)
StopIteration

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "train.py", line 104, in main
model.learn(
File "/workspace/vo_rl/rl_algorithms/ppo.py", line 347, in learn
return super().learn(
File "/workspace/vo_rl/rl_algorithms/on_policy_algorithm.py", line 274, in learn
total_timesteps, callback = self._setup_learn(
File "/venv/lib/python3.8/site-packages/stable_baselines3/common/base_class.py", line 423, in _setup_learn
self._last_obs = self.env.reset() # type: ignore[assignment]
File "/workspace/vo_rl/env/svo_wrapper.py", line 213, in reset
images, gt_poses, new_seq = self.get_images_pose()
File "/workspace/vo_rl/env/svo_wrapper.py", line 89, in get_images_pose
batch = next(self.dataloader_iter)
StopIteration
what's wrong with that

[sujit-168]

same error i meet, , do you have any solution?

[sujit-168]

same error i meet, , do you have any solution?

By adding debug information, located the problem in func

rl_vo/dataloader/tartan_loader.py

Line 132 in 5e36e42

def __getitem__(self, idx):

need to notice code line as follows

• rl_vo/dataloader/tartan_loader.py

  Line 136 in 5e36e42

  self.select_new_traj = np.logical_or(traj_time_idx >= self.nr_samples_per_traj[self.traj_idx],

• rl_vo/dataloader/tartan_loader.py

  Line 150 in 5e36e42

  image_paths = [os.path.join(self.trajectories_paths[self.traj_idx[i]], 'image_left_gray',

• rl_vo/dataloader/tartan_loader.py

  Line 153 in 5e36e42

  images = svo_env.load_image_batch(image_paths, self.num_envs, self.img_h, self.img_w)