Error when trying to re-train

[SM1991CODES]

Hi,

I was trying to re-train on SemanticKITTI and I get the following error during validation. Please help in fixing this.

Traceback (most recent call last):
File "train_cylinder_asym.py", line 167, in
main(args)
File "train_cylinder_asym.py", line 109, in main
iou = per_class_iu(sum(hist_list))
File "/home/sambit/data/code/Cylinder3D/utils/metric_util.py", line 16, in per_class_iu
return np.diag(hist) / (hist.sum(1) + hist.sum(0) - np.diag(hist))
File "<array_function internals>", line 180, in diag
File "/usr/local/lib/python3.8/dist-packages/numpy/lib/twodim_base.py", line 309, in diag
raise ValueError("Input must be 1- or 2-d.")
ValueError: Input must be 1- or 2-d.