According to the instructions in the README, I encountered a ValueError: Buffer dtype mismatch, expected 'int' but got 'long' in train.py at the 20th epoch.

[Leelibrary]

Traceback (most recent call last): File "/home/lab/libr/WakeNet/eval.py", line 100, in evaluate(arg.target_size, File "/home/lab/libr/WakeNet/eval.py", line 85, in evaluate results = data_evaluate(model, target_size, test_path, conf, dataset) File "/home/lab/libr/WakeNet/eval.py", line 43, in data_evaluate dets = im_detect(model, im, target_sizes=target_size, conf=conf) File "/home/lab/libr/WakeNet/utils/detect.py", line 12, in im_detect return single_scale_detect(model, src, target_size=target_sizes[0], use_gpu=use_gpu, conf=conf) File "/home/lab/libr/WakeNet/utils/detect.py", line 36, in single_scale_detect scores, classes, boxes, landmarks = model(im, test_conf=conf) File "/opt/anaconda3/envs/libr/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl return forward_call(*input, **kwargs) File "/home/lab/libr/WakeNet/models/WakeNet.py", line 106, in forward return self.decoder(imgs, anchors_list[-1], cls_score, bbox_pred, land_pred, test_conf=test_conf) File "/home/lab/libr/WakeNet/models/WakeNet.py", line 124, in decoder anchors_nms_idx = nms(torch.cat([bboxes.float(), scores.float()], dim=2)[0, :, :], nms_thresh) File "/home/lab/libr/WakeNet/utils/nms_wrapper.py", line 13, in nms return cpu_nms(dets, thresh) # 加了个int File "cpu_nms.pyx", line 153, in cpu_nms.cpu_nms ValueError: Buffer dtype mismatch, expected 'int' but got 'long'