Traceback (most recent call last):


 "/home/pwwang/.cache/pypoetry/virtualenvs/prog-taq1idOW-py3.8/lib/python3.8/site-packages/astroid/decorators.py", line 32, in cached

return cache[func]
KeyError: <bound method ClassDef.slots of <ClassDef.tqdm l.31 at 0x7f1ccee9f9a0>>

ctdet/seashipTraceback (most recent call last):


 "main.py", line 102, in <module>

main(opt)


 "main.py", line 70, in main

log_dict_train, _ = trainer.train(epoch, train_loader)


 "/home/joe/Desktop/centernet/src/lib/trains/base_trainer.py", line 119, in train

return self.run_epoch('train', epoch, data_loader)


 "/home/joe/Desktop/centernet/src/lib/trains/base_trainer.py", line 61, in run_epoch

for iter_id, batch in enumerate(data_loader):


 "/home/joe/miniconda3/envs/centernet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in __next__

return self._process_next_batch(batch)


 "/home/joe/miniconda3/envs/centernet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch

raise batch.exc_type(batch.exc_msg)
KeyError: 'Traceback (most recent call last):\n

 "/home/joe/miniconda3/envs/centernet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop\n
samples = collate_fn([dataset[i] for i in batch_indices])\n

 "/home/joe/miniconda3/envs/centernet/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in <listcomp>\n
samples = collate_fn([dataset[i] for i in batch_indices])\n

 "/home/joe/Desktop/centernet/src/lib/datasets/sample/ctdet.py", line 102, in __getitem__\n
cls_id = int(self.cat_ids[ann[\'category_id\']])\nKeyError: 0\n'
