yolact训练自己的数据集报错,请问如何解决
loading annotations into memory...
Done (t=0.08s)
creating index...
index created!
loading annotations into memory...
Done (t=0.10s)
creating index...
index created!
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'lat_layers' was found in ScriptModule constants, but it is a non-constan
t submodule. Consider removing it.
" but it is a non-constant {}. Consider removing it.".format(name, hint))
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'pred_layers' was found in ScriptModule constants, but it is a non-consta
nt submodule. Consider removing it.
" but it is a non-constant {}. Consider removing it.".format(name, hint))
C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\jit\_recursive.py:152: UserWarning: 'downsample_layers' was found in ScriptModule constants, but it is a non-
constant submodule. Consider removing it.
" but it is a non-constant {}. Consider removing it.".format(name, hint))
Initializing weights...
Begin training!
Traceback (most recent call last):
File "train.py", line 504, in <module>
train()
File "train.py", line 270, in train
for datum in data_loader:
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 345, in __next__
data = self._next_data()
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 856, in _next_data
return self._process_data(data)
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 881, in _process_data
data.reraise()
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\_utils.py", line 395, in reraise
raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\admin\.conda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\python-object\yolact-train\yolact\data\coco.py", line 94, in __getitem__
im, gt, masks, h, w, num_crowds = self.pull_item(index)
File "D:\python-object\yolact-train\yolact\data\coco.py", line 153, in pull_item
target = self.target_transform(target, width, height)
File "D:\python-object\yolact-train\yolact\data\coco.py", line 42, in __call__
label_idx = self.label_map[label_idx] - 1
KeyError: 0