channge the code in utils.py 235 line original_ind = torch.LongTensor(range(true_class_boxes.size(0)), device='cuda')[true_class_images == this_image][ind] ==> original_ind = torch.tensor(range(true_class_boxes.size(0)), dtype=torch.long, device='cuda')[true_class_images == this_image][ind]