KeyError: 'classes' problem

[Alamnoor]

I am facing the following error. Can you please comment if possible> Python> 3.6

The NVIDIA driver on your system is too old (found version 10010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at /opt/conda/conda-bld/pytorch_1623448233824/work/c10/cuda/CUDAFunctions.cpp:115.) return torch._C._cuda_getDeviceCount() > 0 Dataset sizes: 28152 training, 9384 validation, 9515 testing. Traceback (most recent call last): File "hico.py", line 435, in main(args) File "hico.py", line 115, in main edge_features, node_features, adj_mat, node_labels, sequence_id, det_classes, det_boxes, human_num, obj_num = training_set[0] File "/home/gpnn/src/python/datasets/HICO/hico.py", line 33, in getitem det_classes = data['classes'] KeyError: 'classes'