train on own dataset encounter some error

[He-zl8]

File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/ops/script_ops.py", line 243, in __call__
ret = func(*args)

File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/autograph/impl/api.py", line 309, in wrapper
return func(*args, **kwargs)

File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/data/ops/dataset_ops.py", line 785, in generator_py_func
values = next(generator_state.get_iterator(iterator_id))

File "/content/mydrive/My Drive/ssd-tf2/voc_data.py", line 132, in generate
boxes, labels = self._get_annotation(index, (h, w))

File "/content/mydrive/My Drive/ssd-tf2/voc_data.py", line 105, in _get_annotation
labels.append(self.name_to_idx[name] + 1)

KeyError: 'k'


[[{{node PyFunc}}]]
2020-07-20 13:15:03.262179: W tensorflow/core/kernels/data/generator_dataset_op.cc:103] Error occurred when finalizing GeneratorDataset iterator: Failed precondition: Python interpreter state is not initialized. The process may be terminated.
[[{{node PyFunc}}]]

how to solve this problem.

this is my configuration:
tensorboard 2.2.2
tensorboard-plugin-wit 1.7.0
tensorboardcolab 0.0.22
tensorflow 2.2.0

[LuckyLhy]

Hi，friend. I meet to the same error, have you solved it?