Open
Description
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]
Metadata
Metadata
Assignees
Labels
No labels