noirblack

@liangxi627 Have you sloved it? I got a similar error during testing.

@ayooshkathuria I got the same error as @omg777 at this line: prediction[:, :, :2] += x_y_offset my PyTorch version is 0.3.0.post4, CUDA7.5

@ayooshkathuria I debug predict_transform, and find that, x_y_offset is cuda tensors, prediction is not cuda tensors. I think that most of the darknet is not running on GPU, I add:...

@RyanCV In the function create_modules() add if CUDA: module = module.cuda() before module_list.append(module) I find that @ayooshkathuria had change the module into cuda model in detect.py, so I add this...