hub
hub copied to clipboard
Colab tutoral doesn't work
After this code:
results_per_input = utils.decode_results(detections_batch) best_results_per_input = [utils.pick_best(results, 0.40) for results in results_per_input]
I've got this error:
`RuntimeError Traceback (most recent call last)
2 frames ~/.cache/torch/hub/NVIDIA_DeepLearningExamples_torchhub/PyTorch/Detection/SSD/ssd/utils.py in decode_single(self, bboxes_in, scores_in, criteria, max_output, max_num) 219 _, max_ids = scores_out.sort(dim=0) 220 max_ids = max_ids[-max_output:] --> 221 return bboxes_out[max_ids, :], labels_out[max_ids], scores_out[max_ids] 222 223
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)`