Complex-YOLOv4-Pytorch icon indicating copy to clipboard operation
Complex-YOLOv4-Pytorch copied to clipboard

Fix: Load a model for eval and test

Open SinDongHwan opened this issue 2 years ago • 1 comments

Issue : When load a model for evaluating and testing by default, not get a device information from args.gpu_idx, but read it from the pre-train model(.pth). In a yolov4 case, the device inforamtion is written "cuda:2". So, if have one graphic card, cause error.

Fix : When call torch.load(), I used a parameter(map_location). map_location : a function, torch.device, string or a dict specifying how to remap storage locations.

SinDongHwan avatar Dec 22 '21 06:12 SinDongHwan

This is similar to the PR #18 Can you please resolve the conflict?

maudzung avatar Jul 15 '22 06:07 maudzung