Segmentation error (on training)

[sadbhavsingh16]

No description provided.

[XinyiChu]

do u meet a error like this ?:
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