Colab tutoral doesn't work

[podturkinalex]

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) in ----> 1 results_per_input = utils.decode_results(detections_batch) 2 best_results_per_input = [utils.pick_best(results, 0.40) for results in results_per_input]

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)`