Update util.py - Githubissues

ayooshkathuria / YOLO_v3_tutorial_from_scratch

Accompanying code for Paperspace tutorial series "How to Implement YOLO v3 Object Detector from Scratch"

https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/

2.32k stars 725 forks source link

Update util.py #30

Open zhiweichen0012 opened 5 years ago

zhiweichen0012 commented 5 years ago

RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #4 'other' added：prediction = prediction.cuda()