YOLO_v3_tutorial_from_scratch icon indicating copy to clipboard operation
YOLO_v3_tutorial_from_scratch copied to clipboard

Published 20 hours ago • verified publisher icon ayooshkathuria

Update util.py

Open zhiweichen0012 opened this issue 6 years ago • 0 comments

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

zhiweichen0012 avatar Nov 21 '18 02:11 zhiweichen0012