Index Error

[saja1994]

Please if you can help me. When I tried to run this code using COVID_CT dataset I got this error:
File "main_edge.py", line 34, in main
val_metrics, confusion_matrix = validation(args, model, val_generator, epoch, writer)
File "/content/drive/.shortcut-targets-by-id/104/COVIDNet/trainer/train.py", line 118, in validation
confusion_matrix[t.long(), p.long()] += 1
IndexError: index 968 is out of bounds for dimension 1 with size 2

How can I handle it?!
Thanks